You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The paper mentions the learning rate * 0.1 for the rest of the segmentation head. How is it reflected in the code? I can't find the corresponding part. If I want to make the learning rate of the segmentation head 0.01 and the rest *0.1, I should set self.initial_lr in STUNetTraniner_ft in STUNetTrainer.py to 0.01 or 0.001. My English may be so poor, please forgive me. Thank you very much!
The text was updated successfully, but these errors were encountered:
The paper mentions the learning rate * 0.1 for the rest of the segmentation head. How is it reflected in the code? I can't find the corresponding part. If I want to make the learning rate of the segmentation head 0.01 and the rest *0.1, I should set self.initial_lr in STUNetTraniner_ft in STUNetTrainer.py to 0.01 or 0.001. My English may be so poor, please forgive me. Thank you very much!
The text was updated successfully, but these errors were encountered: