File size: 676 Bytes
4c62147 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
## Quick Start - Training Harmonizer
1. Download the iHarmony4 dataset and put it in the folder `./harmonizer/dataset/`
2. Pre-process the iHarmony4 dataset for training.
We provide the processed Hday2night subset as an example at [this link](https://drive.google.com/drive/folders/1HtrmUlFsT1yIfJ2JkGWwAwFDlv8StD6e?usp=sharing).
You should convert other subsets to the same format for training.
Otherwise, you need to implement new dataset loaders in the file `./harmonizer/data.py` to load datasets with other formats.
3. Run the training script by:
```
cd ./harmonizer
python -m script.train
```
You can config the training arguments in the script. |