Add error for invalid state outputs type in trainer

mosaicml · Jan 15, 2024 · 31d92e8 · 31d92e8
1 parent 63b8ddb
commit 31d92e8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/composer/trainer/trainer.py b/composer/trainer/trainer.py
@@ -2967,8 +2967,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