diff --git a/train.py b/train.py index 6230b95..0f7073d 100644 --- a/train.py +++ b/train.py @@ -1,6 +1,6 @@ learning_rate = 0.65e-4 # 0.65e-4 - 0.4e-4 mini_batch_size = 8 -max_epochs = 120 +max_epochs = 100 from funcs import save_to_file_by_address import json