hli2020 opened this issue 7 years ago · 0 comments
https://github.com/ChenglongChen/pytorch-madrl/blob/44b8f082ffb054c6dabead8971b66a7be4c1f9ff/A2C.py#L56
should add one more line: self.critic.cuda().
Also could you provide some example for running the multi-agent case? Thanks!