refiners/tests/training_utils/mock_config.toml

[models.mock_model]
train = true

[training]
duration = "100:epoch"
seed = 0
device = "cpu"
dtype = "float32"
batch_size = 4
gradient_accumulation = "4:step"
clip_grad_norm = 1.0
evaluation_interval = "5:epoch"
evaluation_seed = 1

[optimizer]
optimizer = "SGD"
learning_rate = 1
momentum = 0.9

[scheduler]
scheduler_type = "ConstantLR"
update_interval = "1:step"
warmup = "20:step"

[dropout]
dropout = 0.0
add basic unit test for training_utils 2024-01-14 14:06:48 +00:00			`[models.mock_model]`
			`train = true`

			`[training]`
			`duration = "100:epoch"`
			`seed = 0`
make device and dtype work in Trainer class 2024-02-06 21:39:38 +00:00			`device = "cpu"`
			`dtype = "float32"`
add basic unit test for training_utils 2024-01-14 14:06:48 +00:00			`batch_size = 4`
			`gradient_accumulation = "4:step"`
			`clip_grad_norm = 1.0`
			`evaluation_interval = "5:epoch"`
			`evaluation_seed = 1`

			`[optimizer]`
			`optimizer = "SGD"`
			`learning_rate = 1`
			`momentum = 0.9`

			`[scheduler]`
			`scheduler_type = "ConstantLR"`
			`update_interval = "1:step"`
			`warmup = "20:step"`

			`[dropout]`
			`dropout = 0.0`