diff --git a/train.py b/train.py index 1e86e332e0c186bdedd44366816dac266513f8dc..ea3c68554307d3e7934cfa812d7d89e62684a5ed 100644 --- a/train.py +++ b/train.py @@ -177,7 +177,7 @@ if __name__ == "__main__": # 提示OOM或者显存不足请调小Batch_size #------------------------------------------------------# if True: - lr = 3e-4 + lr = 5e-4 Batch_size = 32 Init_Epoch = 0 Freeze_Epoch = 50 @@ -211,7 +211,7 @@ if __name__ == "__main__": lr_scheduler.step(val_loss) if True: - lr = 5e-5 + lr = 1e-4 Batch_size = 16 Freeze_Epoch = 50 Unfreeze_Epoch = 100