Yuchan
commited on
Update Mo.py
Browse files
Mo.py
CHANGED
|
@@ -226,13 +226,6 @@ def smoothed_loss_keras(y_true, y_pred, eps=0.1):
|
|
| 226 |
per_tok = per_tok * mask
|
| 227 |
return tf.reduce_sum(per_tok) / (tf.reduce_sum(mask) + 1e-8)
|
| 228 |
|
| 229 |
-
def masked_accuracy(y_true, y_pred):
|
| 230 |
-
y_true = tf.cast(y_true, tf.int32)
|
| 231 |
-
mask = tf.cast(tf.not_equal(y_true, pad_id), tf.float32)
|
| 232 |
-
pred_id = tf.argmax(y_pred, axis=-1, output_type=tf.int32)
|
| 233 |
-
acc = tf.cast(tf.equal(y_true, pred_id), tf.float32) * mask
|
| 234 |
-
return tf.reduce_sum(acc) / (tf.reduce_sum(mask) + 1e-8)
|
| 235 |
-
|
| 236 |
def masked_perplexity(y_true, y_pred, eps=0.1):
|
| 237 |
y_true = tf.cast(y_true, tf.int32)
|
| 238 |
mask = tf.cast(tf.not_equal(y_true, pad_id), tf.float32)
|
|
@@ -255,7 +248,7 @@ with strategy.scope():
|
|
| 255 |
model.summary()
|
| 256 |
|
| 257 |
optimizer = tf.keras.optimizers.Adam(1e-4, beta_1=0.9, beta_2=0.95, epsilon=1e-8, clipnorm=1.0)
|
| 258 |
-
model.compile(optimizer=optimizer, loss=smoothed_loss_keras, metrics=[
|
| 259 |
|
| 260 |
# 학습
|
| 261 |
history = model.fit(dist_dataset, epochs=1, verbose=1)
|
|
|
|
| 226 |
per_tok = per_tok * mask
|
| 227 |
return tf.reduce_sum(per_tok) / (tf.reduce_sum(mask) + 1e-8)
|
| 228 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 229 |
def masked_perplexity(y_true, y_pred, eps=0.1):
|
| 230 |
y_true = tf.cast(y_true, tf.int32)
|
| 231 |
mask = tf.cast(tf.not_equal(y_true, pad_id), tf.float32)
|
|
|
|
| 248 |
model.summary()
|
| 249 |
|
| 250 |
optimizer = tf.keras.optimizers.Adam(1e-4, beta_1=0.9, beta_2=0.95, epsilon=1e-8, clipnorm=1.0)
|
| 251 |
+
model.compile(optimizer=optimizer, loss=smoothed_loss_keras, metrics=[masked_perplexity])
|
| 252 |
|
| 253 |
# 학습
|
| 254 |
history = model.fit(dist_dataset, epochs=1, verbose=1)
|