refiners/tests/training_utils/mock_config_2_models.toml

29 lines
422 B
TOML
Raw Normal View History

2024-02-12 13:17:51 +00:00
[mock_model1]
requires_grad = true
learning_rate = 1e-5
2024-02-12 13:17:51 +00:00
[mock_model2]
requires_grad = true
[clock]
verbose = false
[gradient_clipping]
clip_grad_norm = 1.0
[training]
duration = "100:epoch"
seed = 0
batch_size = 4
gradient_accumulation = "4:step"
evaluation_interval = "5:epoch"
evaluation_seed = 1
[optimizer]
optimizer = "SGD"
learning_rate = 1
[scheduler]
scheduler_type = "ConstantLR"
update_interval = "1:step"