-
- Downloads
Grouped configs and data for default Shenzhen dataset
Created base class for every LightningDataModule, that can be inherited from. For each protocol, the user should create a new class inheriting from BaseDataModule and implement the setup() function, in which the dataset is defined.
Showing
- pyproject.toml 1 addition, 1 deletionpyproject.toml
- src/ptbench/configs/datasets/shenzhen/default.py 0 additions, 15 deletionssrc/ptbench/configs/datasets/shenzhen/default.py
- src/ptbench/data/base_datamodule.py 4 additions, 35 deletionssrc/ptbench/data/base_datamodule.py
- src/ptbench/data/dataset.py 300 additions, 0 deletionssrc/ptbench/data/dataset.py
- src/ptbench/data/shenzhen/__init__.py 370 additions, 4 deletionssrc/ptbench/data/shenzhen/__init__.py
- src/ptbench/data/shenzhen/default.py 38 additions, 0 deletionssrc/ptbench/data/shenzhen/default.py
- src/ptbench/scripts/train.py 4 additions, 5 deletionssrc/ptbench/scripts/train.py
Loading
Please register or sign in to comment