Merge remote-tracking branch 'origin/gcpDebugAdaptive' into gcpDebugA…

…daptive
jerrodparker20 · Apr 2, 2020 · 9c600d7 · 9c600d7
2 parents 19c9ffb + 300f195
commit 9c600d7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -444,7 +444,8 @@ def learn(
 
             learner_outputs, unused_state, mems, curpad_mask, ind_first_done = model(mini_batch, initial_agent_state,
                                                                                      mems=mems)
-            # if mini_batch['done'].any():
+            if mini_batch['done'].any():
+                print('********Should see some return*********')
             #     www = time.time()
             #     torch.save(mini_batch['done'],'./'+str(www)+'mini_batch_done.pt')
             #     print("mini_batch['done'] true at ", www)