Spaces:
Runtime error
Runtime error
File size: 346 Bytes
54c5571 25d090f 54c5571 25d090f 54c5571 |
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 |
seed: 42
epochs: 3
batch_size: 2
num_workers: 0
device: cpu
lr: 1e-4
clip_grad: 1.0
scheduler: none
# Tokenizer path
tokenizer_path: "tokenizer.json"
# Data paths
train_csv: "dataset/train.csv"
val_csv: "dataset/val.csv"
# Image settings
image_height: 64
image_width: 512
max_seq_length: 150
# Saving
output_dir: "checkpoints/"
save_every: 1
|