Datasets can be downloaded here. Extract the archive file and Place the entire folder at the same level as the entire project folder. Or you can put anywhere else and specific the path using the --dataset_root argument.

Pre-trained model can be downloaded here. Extract the archive file and Place the entire folder in the root directory. Or you can put anywhere else and specific the path using the --model_path argument.

The basic running script is located in the src directory, and you can use any additional arguments located in src/args.py to change the program's behavior.

Wandb is used to track running, the mode can be set to offline(default), online and disabled, if --wandb_mode is set to online, it requires user's wandb key.

Sign up for an account at https://wandb.ai and copy your user's key to wandb_api.key in the root directory.