File size: 713 Bytes
3a2e5f0
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# =============================================================================
# configs/train/debug.yaml — fast end-to-end smoke run.
# -----------------------------------------------------------------------------
# Used by CI to verify the training pipeline imports and steps once without
# OOMing or producing NaNs. Loads on top of base.yaml so only the changed
# fields need to be listed.
#
#   python -m scripts.train --config configs/base.yaml --override configs/train/debug.yaml
# =============================================================================

data:
  sample_size: 64               # Just enough captions to fill one batch

train:
  epochs: 1
  batch_size: 8
  buffer_size: 16
  seed: 0