Change default batch size to 2

2024-02-09 16:01:30 +09:00
parent 5876624f01
commit 527908626b
4 changed files with 8 additions and 6 deletions
--- a/configs/configs_jp_extra.json
+++ b/configs/configs_jp_extra.json
@@ -7,7 +7,7 @@
    "learning_rate": 0.0001,
    "betas": [0.8, 0.99],
    "eps": 1e-9,
-    "batch_size": 24,
+    "batch_size": 2,
    "bf16_run": false,
    "fp16_run": false,
    "lr_decay": 0.99996,
@@ -74,5 +74,5 @@
      "initial_channel": 64
    }
  },
-  "version": "2.1.1-JP-Extra"
+  "version": "2.2-JP-Extra"
 }