Skip to content
This repository has been archived by the owner on Apr 15, 2020. It is now read-only.

Latest commit

 

History

History
19 lines (13 loc) · 786 Bytes

README.md

File metadata and controls

19 lines (13 loc) · 786 Bytes

DATASETS

Running the code

python3 trainval.py -e AWSDR_SDx2_edgar -sb ./directory/to/save/logs -j 0 -r 1 # -j is for cluster, -r is for reset or start from last checkpoint

The experiment you want to run is defined with -e

Defining an experiment

Modify: exp_configs/edgar.py

Dependencies

pip install --upgrade git+https://github.com/ElementAI/haven

TODO Edgar

  • Modify models/kornia_trainer.py, add the corresponding losses (commented in the code)