# The main.py is based on https://github.com/pytorch/examples/tree/master/imagenet (under a BSD-3-Clause license).

export CUDA_VISIBLE_DEVICES=0,1,2,3 && python -u main.py --arch resnet50 --optim rstbfgs --dist-url 'tcp://127.0.0.1:9000' --dist-backend 'nccl' --multiprocessing-distributed --world-size 1 --rank 0 --lr 0.1 --momentum 0 --damp1 1 --damp2 1 --beta 0.5 --alpha 1 --weight_decay 0.0001 --seed 1  data-dir > log_rstbfgs_resnet50 &&  python -u showprec.py

