neural-os

Runtime error

da03 commited on Apr 26

Commit

8ce373e

1 Parent(s): e519ed0

.

Files changed (1) hide show

main.py CHANGED Viewed

@@ -131,6 +131,7 @@ def prepare_model_inputs(
             print ('DEBUG MODE_2, REMOVING HIDDEN STATES')
             if 'hidden_states' in inputs:
                 del inputs['hidden_states']
     return inputs
 @torch.no_grad()

             print ('DEBUG MODE_2, REMOVING HIDDEN STATES')
             if 'hidden_states' in inputs:
                 del inputs['hidden_states']
+    print (f'Time step: {time_step}')
     return inputs
 @torch.no_grad()