Tianshou

History

Robust conversion from/to numpy/pytorch (#63 )

* Enable to convert Batch data back to torch.

* Add torch converter to collector.

* Fix

* Move to_numpy/to_torch convert in dedicated utils.py.

* Use to_numpy/to_torch to convert arrays.

* fix lint

* fix

* Add unit test to check Batch from/to numpy.

* Fix Batch over Batch.

Co-authored-by: Alexis Duburcq <alexis.duburcq@wandercraft.eu>

2020-05-29 20:45:21 +08:00

__init__.py

enable null buffer in test collector

2020-04-20 11:50:18 +08:00

base.py

Robust conversion from/to numpy/pytorch (#63 )

2020-05-29 20:45:21 +08:00