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

Added Colab default values

Browse Source
This commit is contained in:
Christian Risi 2025-10-12 00:15:54 +02:00
parent e579e1c88b
commit 46ee6055ec
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
Playgrounds/nanosocrates-train-experiment-2.py
View File

@ -46,8 +46,8 @@ NUMBER_OF_BLOCKS = 4
MAX_EPOCHS = int(1e3) MAX_EPOCHS = int(1e3)
PRETRAIN_EPOCHS = int(300) PRETRAIN_EPOCHS = int(300)
WARMUP_EPOCHS = int(4e3) WARMUP_EPOCHS = int(4e3)
MINI_BATCH_SIZE = 100 MINI_BATCH_SIZE = 300
VALIDATION_STEPS = 50 VALIDATION_STEPS = 25
CHECKPOINT_STEPS = VALIDATION_STEPS * 4 CHECKPOINT_STEPS = VALIDATION_STEPS * 4
PATIENCE = 4 PATIENCE = 4
CURRENT_EPOCH = 0 CURRENT_EPOCH = 0