Update train.py

0f9d2c6f · Bubbliiiing · GitHub · 41ed91c0 · 0f9d2c6f
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

train.py train.py +1 -1

未找到文件。
--- a/train.py
+++ b/train.py
@@ -302,7 +302,7 @@ if __name__ == "__main__":
    if True:
        Init_epoch = 0
        Freeze_epoch = 50
-        batch_size = 16
+        batch_size = 8
        learning_rate_base = 1e-3

        if Cosine_scheduler: