diff --git a/deepmd/loggers/training.py b/deepmd/loggers/training.py index 555ab32622..aada063640 100644 --- a/deepmd/loggers/training.py +++ b/deepmd/loggers/training.py @@ -12,7 +12,7 @@ def format_training_message( eta: int | None = None, ) -> str: """Format a training message.""" - msg = f"batch {batch:7d}: total wall time = {wall_time:.2f} s" + msg = f"Batch {batch:7d}: total wall time = {wall_time:.2f} s" if isinstance(eta, int): msg += f", eta = {datetime.timedelta(seconds=int(eta))!s}" return msg @@ -49,7 +49,7 @@ def format_training_message_per_task( # sort rmse rmse = dict(sorted(rmse.items())) msg = ( - f"batch {batch:7d}: {task_name}" + f"Batch {batch:7d}: {task_name}" f"{', '.join([f'{kk} = {vv:8.2e}' for kk, vv in rmse.items()])}" f"{lr}" ) diff --git a/deepmd/pd/entrypoints/main.py b/deepmd/pd/entrypoints/main.py index f66a2225ee..0cfe610fb5 100644 --- a/deepmd/pd/entrypoints/main.py +++ b/deepmd/pd/entrypoints/main.py @@ -219,8 +219,8 @@ def get_backend_info(self) -> dict: op_info = {} return { "Backend": "Paddle", - "PD ver": f"v{paddle.__version__}-g{paddle.version.commit[:11]}", - "Enable custom OP": False, + "PD Ver": f"v{paddle.__version__}-g{paddle.version.commit[:11]}", + "Custom OP Enabled": False, **op_info, } diff --git a/deepmd/pt/entrypoints/main.py b/deepmd/pt/entrypoints/main.py index 71a358173e..46ad8a6cd0 100644 --- a/deepmd/pt/entrypoints/main.py +++ b/deepmd/pt/entrypoints/main.py @@ -239,16 +239,16 @@ def get_backend_info(self) -> dict: """Get backend information.""" if ENABLE_CUSTOMIZED_OP: op_info = { - "build with PT ver": GLOBAL_CONFIG["pt_version"], - "build with PT inc": GLOBAL_CONFIG["pt_include_dir"].replace(";", "\n"), - "build with PT lib": GLOBAL_CONFIG["pt_libs"].replace(";", "\n"), + "Built with PT Ver": GLOBAL_CONFIG["pt_version"], + "Built with PT Inc": GLOBAL_CONFIG["pt_include_dir"].replace(";", "\n"), + "Built with PT Lib": GLOBAL_CONFIG["pt_libs"].replace(";", "\n"), } else: op_info = {} return { "Backend": "PyTorch", - "PT ver": f"v{torch.__version__}-g{torch.version.git_version[:11]}", - "Enable custom OP": ENABLE_CUSTOMIZED_OP, + "PT Ver": f"v{torch.__version__}-g{torch.version.git_version[:11]}", + "Custom OP Enabled": ENABLE_CUSTOMIZED_OP, **op_info, } diff --git a/deepmd/tf/train/run_options.py b/deepmd/tf/train/run_options.py index e34244abbf..88ec17e188 100644 --- a/deepmd/tf/train/run_options.py +++ b/deepmd/tf/train/run_options.py @@ -67,10 +67,10 @@ def get_backend_info(self) -> dict: """Get backend information.""" return { "Backend": "TensorFlow", - "TF ver": tf.version.GIT_VERSION, - "build with TF ver": TF_VERSION, - "build with TF inc": GLOBAL_CONFIG["tf_include_dir"].replace(";", "\n"), - "build with TF lib": GLOBAL_CONFIG["tf_libs"].replace(";", "\n"), + "TF Ver": tf.version.GIT_VERSION, + "Built with TF Ver": TF_VERSION, + "Built with TF Inc": GLOBAL_CONFIG["tf_include_dir"].replace(";", "\n"), + "Built with TF Lib": GLOBAL_CONFIG["tf_libs"].replace(";", "\n"), } def get_device_name(self) -> str | None: diff --git a/deepmd/utils/data_system.py b/deepmd/utils/data_system.py index c9edfc2298..dd60d9a7e0 100644 --- a/deepmd/utils/data_system.py +++ b/deepmd/utils/data_system.py @@ -715,9 +715,9 @@ def print_summary( # width 65 sys_width = 42 log.info( - f"---Summary of DataSystem: {name:13s}-----------------------------------------------" + f"---Summary of DataSystem: {name.capitalize():13s}-----------------------------------------------" ) - log.info("found %d system(s):", nsystems) + log.info("Found %d System(s):", nsystems) log.info( "%s %6s %6s %6s %9s %3s", _format_name_length("system", sys_width), diff --git a/deepmd/utils/summary.py b/deepmd/utils/summary.py index f052f67699..72aa61bc1d 100644 --- a/deepmd/utils/summary.py +++ b/deepmd/utils/summary.py @@ -47,13 +47,13 @@ class SummaryPrinter(ABC): ) BUILD: ClassVar = { - "installed to": "\n".join(deepmd.__path__), - "source": GLOBAL_CONFIG["git_summ"], - "source branch": GLOBAL_CONFIG["git_branch"], - "source commit": GLOBAL_CONFIG["git_hash"], - "source commit at": GLOBAL_CONFIG["git_date"], - "use float prec": global_float_prec, - "build variant": GLOBAL_CONFIG["dp_variant"], + "Installed to": "\n".join(deepmd.__path__), + "Source": GLOBAL_CONFIG["git_summ"], + "Source Branch": GLOBAL_CONFIG["git_branch"], + "Source Commit": GLOBAL_CONFIG["git_hash"], + "Source Commit at": GLOBAL_CONFIG["git_date"], + "Float Precision": global_float_prec.capitalize(), + "Build Variant": GLOBAL_CONFIG["dp_variant"].upper(), } def __call__(self) -> None: @@ -64,14 +64,14 @@ def __call__(self) -> None: if len(nodelist) > 1: build_info.update( { - "world size": str(len(nodelist)), - "node list": ", ".join(set(nodelist)), + "World Size": str(len(nodelist)), + "Node List": ", ".join(set(nodelist)), } ) build_info.update( { - "running on": nodename, - "computing device": self.get_compute_device(), + "Running on": nodename, + "Computing Device": self.get_compute_device().upper(), } ) device_name = self.get_device_name() @@ -84,13 +84,13 @@ def __call__(self) -> None: env_value = os.environ.get("HIP_VISIBLE_DEVICES", "unset") build_info["HIP_VISIBLE_DEVICES"] = env_value if self.is_built_with_cuda() or self.is_built_with_rocm(): - build_info["Count of visible GPUs"] = str(self.get_ngpus()) + build_info["Visible GPU Count"] = str(self.get_ngpus()) intra, inter = get_default_nthreads() build_info.update( { - "num_intra_threads": str(intra), - "num_inter_threads": str(inter), + "Num Intra Threads": str(intra), + "Num Inter Threads": str(inter), } ) # count the maximum characters in the keys and values