Skip to content

Commit

Permalink
some modification
Browse files Browse the repository at this point in the history
  • Loading branch information
xingzhaohu authored and xingzhaohu committed Apr 1, 2020
1 parent 6c08b87 commit a99b319
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/写诗_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ def iteration(self, epoch, dataloader, train=True):
step = 0
for token_ids, token_type_ids, target_ids in tqdm(dataloader,position=0, leave=True):
step += 1
if step % 100 == 0:
if step % 3000 == 0:
self.bert_model.eval()
test_data = ["观棋##五言绝句", "题西林壁##七言绝句", "长安早春##五言律诗"]
for text in test_data:
Expand Down

0 comments on commit a99b319

Please sign in to comment.