diff --git a/examples/task_relation_extraction.py b/examples/task_relation_extraction.py
index d3eedc4a..967b62c9 100644
--- 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)