Skip to content

Latest commit

 

History

History
14 lines (7 loc) · 325 Bytes

README.md

File metadata and controls

14 lines (7 loc) · 325 Bytes

Pay attention to declare the num classes as needed in my_utils.py and also in the train.py

Loading the data:

python ./utils/my_utils.py

training:

python ./train.py --config ./config.json

validating:

python ./test.py --resume ./saved/models/new_model/0211_113927/checkpoint-epoch30.pth --c ./config.json