Open KK666-AI opened 4 years ago
Thanks. It works.
But it seems the wrapped environment with num_processes=1 is much slower than the original gym environment.
@lihuiknight can you provide parameters that you use and specs of your hardware?
I test this program on ubuntu 18.04, cpu=intel i5, 8 CPUs. I compare the running time on CartPole-v1 between standard gym and the wrapped gym. The wrapped gym is much slower than standard gym.
Dear Author,
When I set num_processes=1, some errors occur, i could be a bug.
File "/home/ken/project/pytorch-a2c-ppo-acktr-gail/a2c_ppo_acktr/envs.py", line 232, in step_wait self.stacked_obs[:, self.shape_dim0:] RuntimeError: unsupported operation: some elements of the input tensor and the written-to tensor refer to a single memory location. Please clone() the tensor before performing the operation.