Shmuma / ptan

PyTorch Agent Net: reinforcement learning toolkit for pytorch
MIT License
531 stars 165 forks source link

At least some samples currently broken #4

Closed martinlyons closed 6 years ago

martinlyons commented 6 years ago

Hi,

The DQN_Speedup samples no longer run since the upgrade of torch to 0.4.0. The ptan agent interface no longer accepts the cuda variable as a parameter to the init function

Shmuma commented 6 years ago

Hi!

Thanks for report. Fixed with https://github.com/Shmuma/ptan/commit/84d349225f15a02164f28586b50cf94ee726eacc