From d765b3b14e8b5638d2ea1d898a4fec2948ac7d25 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 27 Sep 2022 18:43:43 +0100 Subject: [PATCH] fix crash --- aimodel/src/subcommands/pretrain_predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aimodel/src/subcommands/pretrain_predict.py b/aimodel/src/subcommands/pretrain_predict.py index 7f1f56b..bf43772 100644 --- a/aimodel/src/subcommands/pretrain_predict.py +++ b/aimodel/src/subcommands/pretrain_predict.py @@ -98,7 +98,7 @@ def run(args): files_done += 1 i_file = 0 handle.close() - logger.write(f"PROGRESS:file {files_done}") + logger.info(f"PROGRESS:file {files_done}") handle = handle_open(filepath_output.replace("+d", str(files_done+1)), write_mode) if output_mode == MODE_JSONL: