PoliBa-DeepLearning/NanoSocrates
2
0
Fork 0
Code Issues 3 Pull Requests 1 Actions Packages Projects Releases Wiki Activity

typo batch size

Browse Source
This commit is contained in:
GassiGiuseppe 2025-10-11 22:11:53 +02:00
parent d8e65bfb8a
commit 14f1c574e7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Playgrounds/nanosocrates-train.py
View File

@ -46,7 +46,7 @@ NUMBER_OF_BLOCKS = 4
MAX_EPOCHS = int(1e3) MAX_EPOCHS = int(1e3)
PRETRAIN_EPOCHS = int(10) PRETRAIN_EPOCHS = int(10)
WARMUP_EPOCHS = int(4e3) WARMUP_EPOCHS = int(4e3)
MINI_BATCH_SIZE = 20 MINI_BATCH_SIZE = 100
VALIDATION_STEPS = 5 VALIDATION_STEPS = 5
CHECKPOINT_STEPS = VALIDATION_STEPS * 4 CHECKPOINT_STEPS = VALIDATION_STEPS * 4
PATIENCE = 4 PATIENCE = 4