File size: 724 Bytes
fe3247f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
checkpointing_steps: null
dataset_name: cifar10
gradient_accumulation_steps: 1
hub_model_id: null
hub_token: null
ignore_mismatched_sizes: false
image_column_name: img
label_column_name: label
learning_rate: 0.1
lr_scheduler_type: cosine
max_eval_samples: null
max_train_samples: null
max_train_steps: 64000
model_name_or_path: MODELS/cifnet-18-tiny
num_train_epochs: 193
num_warmup_steps: 6400
num_workers: 8
output_dir: OUTPUTS/cifnet-18-tiny-lr0.1-baseline
per_device_eval_batch_size: 8
per_device_train_batch_size: 64
push_to_hub: false
report_to: tensorboard
resume_from_checkpoint: null
seed: 42
train_dir: null
train_val_split: 0.15
trust_remote_code: false
validation_dir: null
weight_decay: 0.0
with_tracking: true
|