- Download the iHarmony4 dataset and put it in the folder
./harmonizer/dataset/
- Pre-process the iHarmony4 dataset for training.
We provide the processed Hday2night subset as an example at this link. 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. - Run the training script by:
You can config the training arguments in the script.
cd ./harmonizer python -m script.train
train
Folders and files
Name | Name | Last commit date | ||
---|---|---|---|---|
parent directory.. | ||||