Update train.py

a50d36a4 · Bubbliiiing · GitHub · 0fbfb75e · a50d36a4
隐藏空白更改
内联并排

Showing with 3 addition and 3 deletion

train.py train.py +3 -3

未找到文件。
--- a/train.py
+++ b/train.py
@@ -226,7 +226,7 @@ if __name__ == "__main__":
    # 调整非主干模型first
    if True:
        Init_epoch = 0
-        Freeze_epoch = 25
+        Freeze_epoch = 50
        # batch_size大小，每次喂入多少数据
        batch_size = 8
        # 最大学习率
@@ -265,8 +265,8 @@ if __name__ == "__main__":

    # 解冻后训练
    if True:
-        Freeze_epoch = 25
-        Epoch = 50
+        Freeze_epoch = 50
+        Epoch = 100
        # batch_size大小，每次喂入多少数据
        batch_size = 2