Update run_pretrain.py (#8902)

Author: ZHUI

llm/run_pretrain.py

@@ -585,7 +585,7 @@ def main():
         test_ret = trainer.predict(test_dataset)
         trainer.log_metrics("test", test_ret.metrics)
 
-    if training_args.should_load_dataset:
+    if training_args.do_train and training_args.should_load_dataset:
         effective_tokens_per_second = total_effective_tokens / train_result.metrics["train_runtime"]
         print(f"Effective Tokens per second: {effective_tokens_per_second:.2f}")
         print(f"ips: {effective_tokens_per_second:.2f} tokens/s")