search

rail-berkeley / rlkit

Collection of reinforcement learning algorithms
MIT License
2.5k stars 553 forks source link

This commit fixes #67 #68

Closed vuoristo closed 5 years ago

vuoristo commented 5 years ago

Model parameters trained on GPU are not recoverable on CPU-only computers.

Fix: Use torch.save instead of pickle.dump for saving model parameters.