zcablii / LSKNet

(IJCV2024 & ICCV2023) LSKNet: A Foundation Lightweight Backbone for Remote Sensing
Other
489 stars 40 forks source link

[Feature] Backbone pretrain config #7

Closed TUDelftHao closed 1 year ago

TUDelftHao commented 1 year ago

What's the feature?

Hi, thank you for your amazing work!

I would like to know if you can share the lsknet backbone pre-train config for the ImageNet, like the batch size, optimizer, lr scheduler, learning rate, etc. I want to finetune the backbone and modify the architecture for downstream applications.

Thank you in advance!

Any other context?

No response

zcablii commented 1 year ago

My lsknet backbone pre-train config is: 8 GPU, DROP_PATH=0.1 -b 128 --lr 1e-3 --drop-path $DROP_PATH --amp --epochs 300 --model-ema Based on this code-base: https://github.com/Visual-Attention-Network/VAN-Classification

TUDelftHao commented 1 year ago

My lsknet backbone pre-train config is: 8 GPU, DROP_PATH=0.1 -b 128 --lr 1e-3 --drop-path $DROP_PATH --amp --epochs 300 --model-ema Based on this code-base: https://github.com/Visual-Attention-Network/VAN-Classification

Thank you for your kind reply! I will have a try!