Add error for invalid state outputs type in trainer

mosaicml · Jan 18, 2024 · b945c5a · b945c5a
1 parent a2de08f
commit b945c5a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/composer/trainer/trainer.py b/composer/trainer/trainer.py
@@ -2990,8 +2990,10 @@ def _eval_loop(
                                                 outputs.append(v.cpu())
                                             else:
                                                 outputs.append(v)
-                                    else:
+                                    elif isinstance(self.state.outputs, torch.Tensor):
                                         outputs = self.state.outputs.cpu()
+                                    else:
+                                        raise ValueError('Unsupported type for state.outputs in trainer eval loop.')
                                 else:
                                     outputs = self.state.outputs