Spaces:
Runtime error
Runtime error
Create train.yaml
Browse files- train.yaml +11 -0
train.yaml
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# train.yaml
|
| 2 |
+
|
| 3 |
+
model:
|
| 4 |
+
name: dummy_pix2tex_model
|
| 5 |
+
hidden_size: 256
|
| 6 |
+
|
| 7 |
+
training:
|
| 8 |
+
epochs: 5
|
| 9 |
+
batch_size: 32
|
| 10 |
+
learning_rate: 0.001
|
| 11 |
+
device: cpu
|