search

KU-CVLAB / CAT-Seg

Official Implementation of "CAT-Seg🐱: Cost Aggregation for Open-Vocabulary Semantic Segmentation"
https://ku-cvlab.github.io/CAT-Seg/
MIT License
247 stars 25 forks source link

is swin frozen? #5

Closed yxchng closed 1 year ago

yxchng commented 1 year ago

frozen_stages (https://github.com/KU-CVLAB/CAT-Seg/blob/main/cat_seg/modeling/backbone/swin.py#L545) does not seem to have been set in your code?

Otherwise, where else in your code freezes swin?

hsshin98 commented 1 year ago

We freeze the Swin parameters here (https://github.com/KU-CVLAB/CAT-Seg/blob/7b5e7a53a51c364feacbd68d0adf27ac651f6557/cat_seg/cat_seg_model.py#L77), setting requires_grad = False as backbone_multiplier is given as 0.0 for Swin.