Small batch is better
@@ -23,7 +23,7 @@ DEFAULT_FILE = "data/OpenSubtitles/en/Crime/1994/60_101020_138057_pulp_fiction.x
DATA_DIR = "data/OpenSubtitles/en/"
SAVES_DIR = "saves"
-BATCH_SIZE = 64
+BATCH_SIZE = 32
LEARNING_RATE = 1e-3
MAX_EPOCHES = 1000
MAX_TOKENS = 10
