Spaces:
Runtime error
Runtime error
Delete train.yaml
Browse files- train.yaml +0 -15
train.yaml
DELETED
|
@@ -1,15 +0,0 @@
|
|
| 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
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|