diff --git a/train.py b/train.py index 0cf898e25ade6b7cf241f3df5d38f014eca83831..2efd28313aa91f509ac5608217d77f4f23e9da8d 100644 --- a/train.py +++ b/train.py @@ -60,7 +60,7 @@ if __name__ == "__main__": epoch_size = num_train // Batch_size - for epoch in range(Epoch): + for epoch in range(Start_iter,Epoch): if epoch%10==0: adjust_learning_rate(optimizer,lr,0.95,epoch) loc_loss = 0