Why am I using this method, epoch training 300, but the loss cannot drop to 2

skuley / DIS-Multi-Gpus-Training

DISNet training with multi-gpus

Apache License 2.0

7 stars 1 forks source link

Why am I using this method, epoch training 300, but the loss cannot drop to 2 #1

Closed mimzzzz closed 1 year ago

mimzzzz commented 1 year ago

Run the command as follows： python train_dis.py --tr_im_path dataset/DIS5K/DIS-TR/im --tr_gt_path dataset/DIS5K/DIS-TR/gt --vd_gt_path dataset/DIS5K/DIS-VD/gt --vd_im_path dataset/DIS5K/DIS-VD/im --min_epoch 300 --gt_weight models/custom_gtencoder.ckpt