search

DeepX-inc / machina

Control section: Deep Reinforcement Learning framework
MIT License
279 stars 43 forks source link

Allocate Traj's tensor to cpu #121

Closed rarilurelo closed 5 years ago

rarilurelo commented 5 years ago

Traj's tensor is now allocated to gpu for fast computing. However it is difficult to allocate all tensors of Off-policy traj to gpu.

Solution

  1. allocating traj's tensor to cpu
  2. setting max_length of traj's tensor
rarilurelo commented 5 years ago

196