Views
No views yet
1num_train_epochs: 3
2per_device_train_batch_size: 8
3per_device_eval_batch_size: 4
4gradient_accumulation_steps: 4
5gradient_checkpointing: true
6optim: adamw_bnb_8bit
7warmup_steps: 50
8learning_rate: 1e-4
9weight_decay: 0.01
10logging_steps: 25
11output_dir: "outputs"
12run_name: "idefics2-frozenlake"
13save_strategy: "epoch"
14save_steps: 250
15save_total_limit: 4
16eval_strategy: "steps"
17eval_steps: 250
18bf16: true
19push_to_hub: false
20hub_model_id: "idefics2-frozenlake"
21remove_unused_columns: false
22report_to: "wandb"