Spaces:
Running
Running
Update app.py
Browse files
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 |
-
|
| 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 |
# ===============================
|