Traceback (most recent call last):
File "train_mpe.py", line 157, in
main(sys.argv[1:])
File "train_mpe.py", line 147, in main
total_num_steps = runner.run()
File "D:\off-policy-release\offpolicy\runner\mlp\base_runner.py", line 153, in run
env_info = self.collecter(explore=True, training_episode=True, warmup=False)
File "D:\off-policy-release\offpolicy\runner\mlp\mpe_runner.py", line 145, in shared_collect_rollout
self.train()
File "D:\off-policy-release\offpolicy\runner\mlp\base_runner.py", line 189, in batch_train
train_info, new_priorities, idxes = update(p_id, sample)
File "D:\off-policy-release\offpolicy\algorithms\maddpg\maddpg.py", line 117, in shared_train_policy_on_batch
rewards = to_torch(rewards).to(**self.tpdv).view(-1, 1)
RuntimeError: CUDA error: an illegal memory access was encountered
When I run the maddpg, it encount cuda error
Traceback (most recent call last): File "train_mpe.py", line 157, in
main(sys.argv[1:])
File "train_mpe.py", line 147, in main
total_num_steps = runner.run()
File "D:\off-policy-release\offpolicy\runner\mlp\base_runner.py", line 153, in run
env_info = self.collecter(explore=True, training_episode=True, warmup=False)
File "D:\off-policy-release\offpolicy\runner\mlp\mpe_runner.py", line 145, in shared_collect_rollout
self.train()
File "D:\off-policy-release\offpolicy\runner\mlp\base_runner.py", line 189, in batch_train
train_info, new_priorities, idxes = update(p_id, sample)
File "D:\off-policy-release\offpolicy\algorithms\maddpg\maddpg.py", line 117, in shared_train_policy_on_batch
rewards = to_torch(rewards).to(**self.tpdv).view(-1, 1)
RuntimeError: CUDA error: an illegal memory access was encountered
When I run the maddpg, it encount cuda error