Update task_relation_extraction.py

bojone · Mar 27, 2020 · ea04aad · ea04aad
1 parent e9d9e3f
commit ea04aad
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/examples/task_relation_extraction.py b/examples/task_relation_extraction.py
@@ -185,9 +185,10 @@ def extrac_subject(inputs):
 object_loss = K.sum(K.mean(object_loss, 3), 2)
 object_loss = K.sum(object_loss * mask) / K.sum(mask)
 
+train_model.add_loss(subject_loss + object_loss)
+
 AdamEMA = extend_with_exponential_moving_average(Adam, name='AdamEMA')
 optimizer = AdamEMA(learning_rate=1e-5)
-train_model.add_loss(subject_loss + object_loss)
 train_model.compile(optimizer=optimizer)