|
resume_from_checkpoint_path: None |
|
result_path: './result' |
|
pretrained_model_name_or_path: 'solotimes/lavibe_base' |
|
dataset_name_or_paths: |
|
- 'dataset/lavibe-base' |
|
sort_json_key: False |
|
train_batch_sizes: |
|
- 2 |
|
val_batch_sizes: |
|
- 1 |
|
input_size: |
|
- 1280 |
|
- 960 |
|
max_length: 1502 |
|
align_long_axis: True |
|
num_nodes: 1 |
|
seed: 2022 |
|
lr: 3e-05 |
|
warmup_steps: 446 |
|
num_training_samples_per_epoch: 1188 |
|
max_epochs: 30 |
|
max_steps: -1 |
|
num_workers: 1 |
|
val_check_interval: 1.0 |
|
check_val_every_n_epoch: 3 |
|
gradient_clip_val: 1.0 |
|
verbose: True |
|
exp_name: 'train_lavibe_base' |
|
exp_version: 'lavibe_base' |