PyTorch-Encoding
PyTorch-Encoding copied to clipboard
A problem "--Ir-step 60" occur at runing train_dist.py
Pytorch vision: 1.4.0;
torchvision:0.5.0;
cuda vison:10.2
I run the train_dist.py follow the tutorial of https://hangzhang.org//PyTorch-Encoding/tutorials/texture.html.
I met a problem "train_dist.py: error: unrecognized arguments: --lr-step 60".
I delete ''--Ir-step 60" and add 'lr_step=60' at
scheduler = LR_Scheduler(args.lr_scheduler, base_lr=args.lr, lr_step=60, # added to solve lr_step problem num_epochs=args.epochs, iters_per_epoch=len(train_loader), warmup_epochs=args.warmup_epochs)
Hope author could fix this small problem.