You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Traceback (most recent call last):
File "test/continuous/test_sac_with_il.py", line 145, in <module>
test_sac_with_il()
File "test/continuous/test_sac_with_il.py", line 106, in test_sac_with_il
args.batch_size, stop_fn=stop_fn, save_fn=save_fn, writer=writer)
File "/home/trinkle/github/tianshou-new/tianshou/trainer/offpolicy.py", line 87, in offpolicy_trainer
losses = policy.learn(train_collector.sample(batch_size))
File "/home/trinkle/github/tianshou-new/tianshou/policy/modelfree/sac.py", line 131, in learn
actor_loss.backward()
File "/home/trinkle/.local/lib/python3.6/site-packages/torch/tensor.py", line 198, in backward
torch.autograd.backward(self, gradient, retain_graph, create_graph)
File "/home/trinkle/.local/lib/python3.6/site-packages/torch/autograd/__init__.py", line 100, in backward
allow_unreachable=True) # allow_unreachable flag
RuntimeError: one of the variables needed for gradient computation has been modified by an inplace operation: [torch.cuda.FloatTensor [128, 1]], which is output 0 of TBackward, is at version 2; expected version 1 instead. Hint: enable anomaly detection to find the operation that failed to compute its gradient, with torch.autograd.set_detect_anomaly(True).
The text was updated successfully, but these errors were encountered:
Fail in SACPolicy with torch==1.5.0
The text was updated successfully, but these errors were encountered: