sthalles/SimCLR

keyword arguments to the run.py file

sourabhsugandhi opened this issue · 0 comments

for smooth execution use run.py use following command

BEFORE : $ python run.py -data ./datasets --dataset-name stl10 --log-every-n-steps 100 --epochs 100
AFTER : $ python run.py -data ./datasets -dataset-name stl10 --log-every-n-steps 100 --epochs 100

Or else you can make change in run.py at line number 16 from parser.add_argument('-dataset-name', default='stl10',help='dataset name', choices=['stl10', 'cifar10']) to parser.add_argument('--dataset-name', default='stl10', help='dataset name', choices=['stl10', 'cifar10'])