OpenLab-NLP commited on
Commit
4ad0ec1
·
verified ·
1 Parent(s): 0ca9e64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -2
app.py CHANGED
@@ -35,8 +35,16 @@ if not os.path.exists(TOKENIZER_PATH):
35
  MAX_LEN = 384
36
  EMBED_DIM = 512
37
  LATENT_DIM = 512
38
- DROP_RATE = 0.1
39
-
 
 
 
 
 
 
 
 
40
  # ===============================
41
  # 1️⃣ 토크나이저 로딩
42
  # ===============================
 
35
  MAX_LEN = 384
36
  EMBED_DIM = 512
37
  LATENT_DIM = 512
38
+ BATCH_SIZE = 768 # global batch size (Keras/TPU가 replica-wise로 나눠서 처리)
39
+ EPOCHS = 1
40
+ SHUFFLE_BUFFER = 200000
41
+ LEARNING_RATE = 1e-4
42
+ TEMPERATURE = 0.05
43
+ DROPOUT_AUG = 0.1
44
+ EMBED_DROPOUT = 0.1
45
+ SEED = 42
46
+ DROPOUT_AUG = 0.1
47
+ EMBED_DROPOUT = 0.1
48
  # ===============================
49
  # 1️⃣ 토크나이저 로딩
50
  # ===============================