PyTorch-Encoding icon indicating copy to clipboard operation
PyTorch-Encoding copied to clipboard

A problem "--Ir-step 60" occur at runing train_dist.py

Open Jie2World opened this issue 5 years ago • 0 comments

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.

Jie2World avatar Jul 17 '20 04:07 Jie2World