search

csrhddlam / axial-deeplab

This is a PyTorch re-implementation of Axial-DeepLab (ECCV 2020 Spotlight)
https://arxiv.org/abs/2003.07853
Apache License 2.0
449 stars 69 forks source link

Pretrained weights #20

Open Shujun-He opened 3 years ago

Shujun-He commented 3 years ago

Can you release more pretrained weights in addition to Axial-ResNet-26-S?

csrhddlam commented 3 years ago

Sorry we don't have free GPU resources to run more models, but please let me know if you have run some other models.

Shujun-He commented 3 years ago

I have a 2x3090 machine so I can run some models. Do I just run the train script?

csrhddlam commented 3 years ago

Sounds great!

Yes, just follow the command line in the README, but you might need to linearly scale the learning rate and batch size accordingly.