add save_every for less disk intensive work.

pull/175/head
bobloy 4 years ago
parent b4f20dd7d2
commit 59fd96fc5a

@ -142,6 +142,8 @@ class MovieTrainer(KaggleTrainer):
# # lines_dict = {row[0].strip('"'): row[4] for row in reader_list}
statements_from_file = []
save_every = 50
count = 0
# [characterID of first, characterID of second, movieID, list of utterances]
async for lines in AsyncIter(conv):
@ -167,9 +169,15 @@ class MovieTrainer(KaggleTrainer):
statements_from_file.append(statement)
count += 1
if count >= save_every:
if statements_from_file:
self.chatbot.storage.create_many(statements_from_file)
statements_from_file = []
count = 0
if statements_from_file:
self.chatbot.storage.create_many(statements_from_file)
log.info("Training took", time.time() - start_time, "seconds.")

Loading…
Cancel
Save