Spaces:
Runtime error
Runtime error
Create train.yaml
Browse files- train.yaml +15 -0
train.yaml
ADDED
|
@@ -0,0 +1,15 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
train:
|
| 2 |
+
dataset: dataset_train.csv
|
| 3 |
+
val_dataset: dataset_val.csv
|
| 4 |
+
batch_size: 16
|
| 5 |
+
epochs: 10
|
| 6 |
+
workers: 4
|
| 7 |
+
seed: 42
|
| 8 |
+
mixed_precision: true
|
| 9 |
+
optimizer: adamw
|
| 10 |
+
scheduler: cosine
|
| 11 |
+
lr: 1e-4
|
| 12 |
+
|
| 13 |
+
model:
|
| 14 |
+
checkpoint: "weights/pix2tex-base.pt" # Path to base model, auto-downloads
|
| 15 |
+
max_seq_len: 512
|