Skip to content

Latest commit

 

History

History
10 lines (8 loc) · 443 Bytes

README.md

File metadata and controls

10 lines (8 loc) · 443 Bytes


Training

CUDA_VISIBLE_DEVICES="1,2,3" python3 main.py --epoch 15 --batch_size 1 --nb_classes 20 --data_path /home/anhducnguyen/pascal_2007 --train_csv_path /home/anhducnguyen/pascal_2007/multihot_train.csv --test_csv_path /home/anhducnguyen/pascal_2007/multihot_test.csv