Increase batch_size

jmisilo · Nov 15, 2022 · 92f9623 · 92f9623
1 parent 311dec3
commit 92f9623
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/utils/config.py b/src/utils/config.py
@@ -18,7 +18,7 @@ class Config:
  epochs: int = 150
  lr: int = 3e-3
  k: float = 0.33
- batch_size_exp: int = 6
+ batch_size_exp: int = 7
  ep_len: int = 4
  num_layers: int = 6
  n_heads: int = 16