From f4d1d1d77ec0b3e9cb0cb764f9d8432869c4eea6 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 27 Sep 2022 18:25:45 +0100 Subject: [PATCH] just wh --- aimodel/src/subcommands/pretrain_predict.py | 1 + 1 file changed, 1 insertion(+) diff --git a/aimodel/src/subcommands/pretrain_predict.py b/aimodel/src/subcommands/pretrain_predict.py index 972489a..7f1f56b 100644 --- a/aimodel/src/subcommands/pretrain_predict.py +++ b/aimodel/src/subcommands/pretrain_predict.py @@ -72,6 +72,7 @@ def run(args): output_mode = MODE_TFRECORD if filepath_output.endswith(".tfrecord") or filepath_output.endswith(".tfrecord.gz") else MODE_JSONL logger.info("Output mode is "+("TFRECORD" if output_mode == MODE_TFRECORD else "JSONL")) + logger.info(f"Records per file: {args.records_per_file}") write_mode = "wt" if filepath_output.endswith(".gz") else "w" if output_mode == MODE_TFRECORD: