2 Commits

Author	SHA1	Message	Date
Alexis DUBURCQ	1fce527c77	Fix 'to_tensor' dtype/device forwarding for Batch over Batch. (#68) ... * Fix Batch to_torch method not updating dtype/device of already converted data. * Fix dtype/device to forwarded by to_tensor for Batch over Batch. * Add Unit test to check to_torch dtype/device recursive forwarding. * Batch UT check accessing data using both dict and class style. * Fix utils to_tensor dtype/device forwarding. Add Unit tests. * Fix UT. Co-authored-by: Alexis Duburcq <alexis.duburcq@wandercraft.eu> Co-authored-by: n+e <463003665@qq.com>	2020-05-30 21:40:31 +08:00
Alexis DUBURCQ	8af7196a9a	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