diff --git a/composer/trainer/trainer.py b/composer/trainer/trainer.py index 66140ad72b6..2256bcd1c6b 100644 --- a/composer/trainer/trainer.py +++ b/composer/trainer/trainer.py @@ -2983,6 +2983,13 @@ def _eval_loop( outputs[k] = v.cpu() else: outputs[k] = v + elif isinstance(self.state.outputs, Sequence): + outputs = [] + for v in self.state.outputs: + if isinstance(v, torch.Tensor): + outputs.append(v.cpu()) + else: + outputs.append(v) else: outputs = self.state.outputs.cpu() else: