Update train.py

9ced6c76 · Bubbliiiing · GitHub · f4ae87a4 · 9ced6c76
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

train.py train.py +1 -1

未找到文件。
--- a/train.py
+++ b/train.py
@@ -214,7 +214,7 @@ if __name__ == "__main__":
        Init_epoch = 0
        Freeze_epoch = 25
        # batch_size大小，每次喂入多少数据
-        batch_size = 4
+        batch_size = 8
        # 最大学习率
        learning_rate_base = 1e-3
        if Cosine_scheduler: