File size: 342 Bytes
3f72633 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
DATA_PATH: "./data/test_smi.csv"
OUTPUT_DIR: "./chmbedv2-warmup"
LOSS_TYPE: "barlow"
BATCH_SIZE: 16
EFFECTIVE_BATCH: 64
EPOCHS: 5
MAX_LENGTH: 512
LR: 1e-4 # Higher LR for random init
WEIGHT_DECAY: 0.01
SEED: 42
EVAL_EVERY_N_PERCENT: 25
BEST_BY_HEALTH: true
# Loss-specific params (optional)
BARLOW_LAMBDA: 5.0 |