This commit is contained in:
gouhanke
2026-02-10 15:56:05 +08:00
parent 3c27d6d793
commit 1e95d40bf9
4 changed files with 790 additions and 109 deletions
+1 -1
View File
@@ -5,7 +5,7 @@ defaults:
- _self_
train:
batch_size: 16 # Batch size for training
batch_size: 8 # Batch size for training
lr: 1e-4 # Learning rate
max_steps: 20000 # Maximum training steps
log_freq: 100 # Log frequency (steps)