ChartReformer / train_state.json
Pengyu965's picture
Upload ChartReformer checkpoints (ICDAR 2024) and model card
e7062e9 verified
Raw History Blame Contribute Delete
509 Bytes
{
"run_name": "finetune_new_edits_data",
"max_length": 1024,
"per_device_train_batch_size": 4,
"per_device_eval_batch_size": 4,
"gradient_accumulation_steps": 1,
"learning_rate": 2e-05,
"weight_decay": 1e-05,
"num_warmup_steps": 500,
"train_batch_size": 16,
"num_train_epochs": 20,
"max_steps": 387000,
"checkpointing_steps": "5000",
"epoch": 1,
"global_step": 20000,
"train_loss": 0.03977988660335541,
"eval_loss": 0.05922633409500122,
"current_lr": 1.9874737996766403e-05
}