Update train.py

train.py

@@ -181,7 +181,7 @@ if __name__ == "__main__":
         Freeze_Epoch = 50
 
         optimizer = optim.Adam(net.parameters(), lr=lr)
-        lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.95)
+        lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.92)
 
         if Use_Data_Loader:
             train_dataset = SSDDataset(lines[:num_train], (Config["min_dim"], Config["min_dim"]), True)
@@ -212,7 +212,7 @@ if __name__ == "__main__":
         Unfreeze_Epoch = 100
 
         optimizer = optim.Adam(net.parameters(), lr=lr)
-        lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.95)
+        lr_scheduler = optim.lr_scheduler.StepLR(optimizer,step_size=1,gamma=0.92)
 
         if Use_Data_Loader:
             train_dataset = SSDDataset(lines[:num_train], (Config["min_dim"], Config["min_dim"]), True)