Spaces:
Runtime error
Runtime error
Commit ·
9203994
1
Parent(s): 52db32e
epochs
Browse files- train_model.py +1 -1
train_model.py
CHANGED
|
@@ -35,7 +35,7 @@ CHAR_CNN_OUT = 50 # Increased from 30
|
|
| 35 |
BBOX_DIM = 128 # Increased from 100
|
| 36 |
HIDDEN_SIZE = 768 # Increased from 512 to match LayoutLM dimension
|
| 37 |
BATCH_SIZE = 8
|
| 38 |
-
EPOCHS =
|
| 39 |
LR = 5e-4 # Decreased from 1e-3 for more stable training
|
| 40 |
BBOX_NORM_CONSTANT = 1000.0
|
| 41 |
CHUNK_SIZE = 450
|
|
|
|
| 35 |
BBOX_DIM = 128 # Increased from 100
|
| 36 |
HIDDEN_SIZE = 768 # Increased from 512 to match LayoutLM dimension
|
| 37 |
BATCH_SIZE = 8
|
| 38 |
+
EPOCHS = 1 # Increased from 30
|
| 39 |
LR = 5e-4 # Decreased from 1e-3 for more stable training
|
| 40 |
BBOX_NORM_CONSTANT = 1000.0
|
| 41 |
CHUNK_SIZE = 450
|