1af1dee919
* first version of torch data adapter * fix missing attribute * 1. Rename TorchDatasetAdapter to TorchDataLoaderAdapter 2. Replace dataset with dataloader 3. Add checks for partial batch size for tensordataset 4. Do inline import * add tests for torch data adapter * remove class_weight argument * add torch to the package list * make sure torch is installed for CPU only * update link for torch cpu install * fix torch cpu install download link * fix bug in partial size calculation * update tests * add get_tf_dataset method * fix link for torch * try index link flag * fix torch whl link * cpu only install * add torch cpu * fix requirements.txt * add note * inline torch import and modify tests |
||
|---|---|---|
| .. | ||
| __init__.py | ||
| array_data_adapter_test.py | ||
| array_data_adapter.py | ||
| data_adapter_utils.py | ||
| data_adapter.py | ||
| generator_data_adapter_test.py | ||
| generator_data_adapter.py | ||
| py_dataset_adapter_test.py | ||
| py_dataset_adapter.py | ||
| tf_dataset_adapter_test.py | ||
| tf_dataset_adapter.py | ||
| torch_data_adapter_test.py | ||
| torch_data_adapter.py | ||