We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3f59eff commit e2c6912Copy full SHA for e2c6912
PaddleNLP/language_model/train.py
@@ -287,9 +287,9 @@ def eval(data):
287
288
if epoch_id == max_epoch - 1 and args.enable_ce:
289
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]))
+ print("ptblm\tlstm_language_model_%s_duration_card%d\t%s" %
+ (args.rnn_model, card_num, total_time / max_epoch))
+ print("ptblm\tlstm_language_model_%s_loss_card%d\t%s" % (args.rnn_model, card_num, ppl[0]))
293
294
model_path = os.path.join("model_new/", str(epoch_id))
295
if not os.path.isdir(model_path):
0 commit comments