md.simulation_hooks.TensorBoardLogger

class md.simulation_hooks.TensorBoardLogger(*args: Any, **kwargs: Any)[source]

TensorBoard logging hook for the properties of the replicas, as well as of the corresponding centroids for each molecule in the system container.

Parameters:

  • log_file (str) – Path to the TensorBoard file.

  • every_n_steps (int) – Frequency with which data is logged to TensorBoard.