PyTorch Implementation

1. Dependencies

• python = 3.8.0
• pytorch = 1.12.0
• numpy = 1.24.3
• scipy = 1.10.1
• scikit-learn = 1.3.0
• PyYAML = 6.0.1

2. Training Configuration

The hyperparameters needed in training can be set via the configuration file: ./configs/config.yaml.

3. Run Training

First of all, we suggest looking through the available command line arguments defined in ./utility/globals.py.

To run the code on CPU

python train.py --data_prefix <dataset_path> --dataset <dataset_name>  --train_config ./configs/config.yaml --gpu -1

To run the code on GPU