Skip to content

Commit e2c6912

Browse files
zhengya01kolinwei
authored andcommitted
Ce lm (#2284)
* update _ce.py * fix ce
1 parent 3f59eff commit e2c6912

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

PaddleNLP/language_model/train.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -287,9 +287,9 @@ def eval(data):
287287

288288
if epoch_id == max_epoch - 1 and args.enable_ce:
289289
card_num = get_cards()
290-
print("ptblm\tlstm_language_model_duration_card%d\t%s" %
291-
(card_num, total_time / max_epoch))
292-
print("ptblm\tlstm_language_model_loss_card%d\t%s" % (card_num, ppl[0]))
290+
print("ptblm\tlstm_language_model_%s_duration_card%d\t%s" %
291+
(args.rnn_model, card_num, total_time / max_epoch))
292+
print("ptblm\tlstm_language_model_%s_loss_card%d\t%s" % (args.rnn_model, card_num, ppl[0]))
293293

294294
model_path = os.path.join("model_new/", str(epoch_id))
295295
if not os.path.isdir(model_path):

0 commit comments

Comments
 (0)