config.yaml

@@ -5,11 +5,11 @@ experiment: "SAN"
 seed: 20211024
 
 # 训练参数
-epoches: 50
+epoches: 240
 batch_size: 8
 workers: 0
 optimizer: Adadelta
-lr: 0.001
+lr: 1.1
 lr_decay: cosine
 step_ratio: 10
 eps: 1e-6
@@ -80,7 +80,7 @@ hybrid_tree:
 # 训练权重存放路径
 optimizer_save: False
 checkpoint_dir: '/model/ckpt'
-finetune: True
+finetune: False
 checkpoint: "/code/checkpoints/56_39.pth"
 
 # tensorboard路径

models/Backbone.py

@@ -247,9 +247,9 @@ class SupConHead(nn.Module):
         super(SupConHead, self).__init__()
 
         self.head = nn.Sequential(
-            nn.Linear(128, 2048),
+            nn.Linear(128, 512),
             nn.ReLU(inplace=True),
-            nn.Linear(2048, 128)
+            nn.Linear(512, 128)
         )
 
     def forward(self, x):