Skip to content

Commit

Permalink
Update README
Browse files Browse the repository at this point in the history
  • Loading branch information
ZHKKKe committed Oct 30, 2022
1 parent 01d7353 commit 1bea960
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/train/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@

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 in the folder `./harmonizer/dataset/iHarmony4/Hday2night`.
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:
Expand Down

0 comments on commit 1bea960

Please sign in to comment.