pytorch-saltnet

pytorch-saltnet copied to clipboard

https://github.com/tugstugi/pytorch-saltnet/blob/a3e63b357f975924e75b5db24ee528c5797f5efb/utils/lr_scheduler.py#L36-L39 The custom `NoamLR` results in same LR in the first two steps, like: ```py warmup_steps = 10 lr = 0.01 ``` - step -- before (current step LR) --...

ShaneTian

谢谢作者的时间，我运行了代码后出现这样的错误

3

(base) C:\YCRS_DATA\YCR_Code\pytorch-saltnet-master>python train.py --vtf --pretrained imagenet --loss-on-center --batch-size 32 --optim adamw --learning-rate 5e-4 --lr-scheduler noam --basenet senet154 --max-epochs 250 --data-fold fold0 --log-dir runs/fold0 --resume runs/fold0/checkpoints/last-checkpoint-fold0.pth Load dataset list_train0_3600: 100%|█████████████████████████████████████████| 3599/3599...

yangtutuaka