esvit icon indicating copy to clipboard operation
esvit copied to clipboard

Loss stops decreasing

Open SarahFrem opened this issue 3 years ago • 1 comments

Hi,

I'm retraining from scratch EsVIT on a custom dataset (1.7M images) with tiny swin, W=14, and a batch size of 64, default lr and wd, and the following hp --teacher_temp 0.04
--warmup_teacher_temp 0.03
--momentum_teacher 0.9996
--warmup_epochs 10
--warmup_teacher_temp_epochs 30
--use_dense_prediction True
--use_fp16 True
--out_dim 65536
--epochs 300 \

The loss does not decrease from epoch 70 onwards.

Which hp would you recommend tuning now resuming from let's say epoch 70 ?

Thanks

image

SarahFrem avatar Aug 29 '22 12:08 SarahFrem

Hi,do you have ckpt and train logs , can you share with me ? I got an error ,when I download them.

sym0926 avatar Aug 14 '24 11:08 sym0926