Some questions about the setting of initial_ir

[xl-lei]

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!