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: