Segmentation.md
March 13, 2024 ยท View on GitHub
Plain VisionLLaMA
VisionLLaMA-B with 800 epochs using MAE.
cd mmseg
bash ./tools/dist_train.sh configs/mae/lama-base_upernet_8xb2-800-amp-160k_ade20k-512x512.py 8 --amp &> lama-base_upernet_8xb2-800-amp-160k_ade20k-512x512.log
VisionLLaMA-B with 1600 epochs using MAE.
cd mmseg
bash ./tools/dist_train.sh configs/mae/lama-base_upernet_8xb2-1600-amp-160k_ade20k-512x512.py 8 --amp &> lama-base_upernet_8xb2-1600-amp-160k_ade20k-512x512.log
Pyramid VisionLLaMA
Pyramid VisionLLaMA-B
cd mmseg
bash ./tools/dist_train.sh configs/twins/twins_svt_lama_as-b_uperhead_8xb2-160k_ade20k-512x512.py 8 --amp &> twins_svt_lama_as-b_uperhead_8xb2-160k_ade20k-512x512.log
Pyramid VisionLLaMA-L
cd mmseg
bash ./tools/dist_train.sh configs/twins/twins_svt_lama_as-l_uperhead_8xb2-160k_ade20k-512x512.py 8 --amp &> twins_svt_lama_as-l_uperhead_8xb2-160k_ade20k-512x512.log
Acknowledgement
Our code is based on mmsegmentation and Twins. Thanks for their great work. Specifically, we start from this commit id.