deepmd.loggers.training

Module Contents

Functions

format_training_message(batch, wall_time)	Format a training message.
format_training_message_per_task(batch, task_name, ...)

deepmd.loggers.training.format_training_message(batch: int, wall_time: float)

Format a training message.

deepmd.loggers.training.format_training_message_per_task(batch: int, task_name: str, rmse: Dict[str, float], learning_rate: float | None)