Update train.py

train.py

@@ -25,7 +25,7 @@ if __name__ == "__main__":
     #   先大学习率
     #   后小学习率
     # ------------------------------------#
-    lr = 1e-5
+    lr = 1e-4
     freeze_lr = 1e-5
     Cuda = True
 
@@ -164,4 +164,4 @@ if __name__ == "__main__":
                 print('iter:' + str(iteration) + '/' + str(epoch_size) + ' || Loc_Loss: %.4f || Conf_Loss: %.4f ||' % (loc_loss/(iteration+1),conf_loss/(iteration+1)), end=' ')
 
             print('Saving state, iter:', str(epoch+1))
-            torch.save(model.state_dict(), 'logs/Epoch%d-Loc%.4f-Conf%.4f.pth'%((epoch+1),loc_loss/(iteration+1),conf_loss/(iteration+1)))
\ No newline at end of file
+            torch.save(model.state_dict(), 'logs/Epoch%d-Loc%.4f-Conf%.4f.pth'%((epoch+1),loc_loss/(iteration+1),conf_loss/(iteration+1)))