Hugging Face's logo

Spaces:
heerjtdev
/
LSTM-CRF_Train
Sleeping

App Files Community
aagamjtdev commited on
Commit
7169978
·
1 Parent(s): dbfc25b

restating the space

Browse files
Files changed (1) hide show
  1. train_model.py +1 -1
train_model.py CHANGED
@@ -33,7 +33,7 @@ CHAR_CNN_OUT = 50 # Increased from 30
33
  BBOX_DIM = 128 # Increased from 100
34
  HIDDEN_SIZE = 768 # Increased from 512 to match LayoutLM dimension
35
  BATCH_SIZE = 8
36
- EPOCHS = 1 # Increased from 30
37
  LR = 5e-4 # Decreased from 1e-3 for more stable training
38
  BBOX_NORM_CONSTANT = 1000.0
39
  CHUNK_SIZE = 450
 
33
  BBOX_DIM = 128 # Increased from 100
34
  HIDDEN_SIZE = 768 # Increased from 512 to match LayoutLM dimension
35
  BATCH_SIZE = 8
36
+ EPOCHS = 8 # Increased from 30
37
  LR = 5e-4 # Decreased from 1e-3 for more stable training
38
  BBOX_NORM_CONSTANT = 1000.0
39
  CHUNK_SIZE = 450