RealBasicVSR icon indicating copy to clipboard operation
RealBasicVSR copied to clipboard

loss is nan

Open zlu1994 opened this issue 2 years ago • 1 comments

"loss_pix" and "loss_clean" is nan every time I train for a while I follow the instructions below: Put the original REDS dataset in ./data Run the following command: python crop_sub_images.py --data-root ./data/REDS --scales 4

and training model follow the instructions mim train mmedit configs/realbasicvsr_wogan_c64b20_2x30x8_lr1e-4_300k_reds.py --gpus 2 --launcher pytorch

zlu1994 avatar Jun 08 '22 02:06 zlu1994

have you tried a smaller lr?

zhu2bowen avatar Nov 27 '23 02:11 zhu2bowen