Hugging Face's logo

Thouph
/
test_temp

Model card Files Community
Thouph commited on
Commit
9f65be4
·
1 Parent(s): 187c12b

Upload train_k.py

Browse files
Files changed (1) hide show
  1. train_k.py +2 -2
train_k.py CHANGED
@@ -37,8 +37,8 @@ else:
37
  class config :
38
  ENCODER = "google/vit-base-patch16-224"
39
  DECODER = "gpt2"
40
- TRAIN_BATCH_SIZE = 4#8
41
- VAL_BATCH_SIZE = 4#8
42
  VAL_EPOCHS = 1
43
  LR = 5e-5
44
  SEED = 42
 
37
  class config :
38
  ENCODER = "google/vit-base-patch16-224"
39
  DECODER = "gpt2"
40
+ TRAIN_BATCH_SIZE = 64#8
41
+ VAL_BATCH_SIZE = 64#8
42
  VAL_EPOCHS = 1
43
  LR = 5e-5
44
  SEED = 42