diff --git a/train_with_tensorboard.py b/train_with_tensorboard.py
index 400c74c00ef4c66e50ea306b918897fd58766de9..eba0c3325d12ac742268752830c220a6183856e4 100644
--- a/train_with_tensorboard.py
+++ b/train_with_tensorboard.py
@@ -196,7 +196,7 @@ if __name__ == "__main__":
         lr = 1e-3
         Batch_size = 4
         Init_Epoch = 0
-        Freeze_Epoch = 25
+        Freeze_Epoch = 50
         
         optimizer = optim.Adam(net.parameters(),lr,weight_decay=5e-4)
         if Cosine_lr:
@@ -232,8 +232,8 @@ if __name__ == "__main__":
     if True:
         lr = 1e-4
         Batch_size = 2
-        Freeze_Epoch = 25
-        Unfreeze_Epoch = 50
+        Freeze_Epoch = 50
+        Unfreeze_Epoch = 100
 
         optimizer = optim.Adam(net.parameters(),lr,weight_decay=5e-4)
         if Cosine_lr: