Skip to content

Commit

Permalink
Fixing minor issues
Browse files Browse the repository at this point in the history
  • Loading branch information
nerkulec committed Nov 19, 2024
1 parent cab6b12 commit ae2e0ed
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
2 changes: 1 addition & 1 deletion docs/source/advanced_usage/trainingmodel.rst
Original file line number Diff line number Diff line change
Expand Up @@ -195,7 +195,7 @@ By default, the same number of snapshots as had been provided will be created
(if possible).

Logging metrics during training
******************
*******************************

Training progress in MALA can be visualized via tensorboard or wandb, as also shown
in the file ``advanced/ex03_tensor_board``. Simply select a logger prior to training as
Expand Down
6 changes: 3 additions & 3 deletions mala/common/parameters.py
Original file line number Diff line number Diff line change
Expand Up @@ -713,7 +713,7 @@ class ParametersRunning(ParametersBase):
validation_metrics : list
List of metrics to be used for validation. Default is ["ldos"].
Possible options are:
- "ldos": Loss on the LDOS.
- "ldos": MSE of the LDOS.
- "band_energy": Band energy.
- "band_energy_actual_fe": Band energy computed with ground truth Fermi energy.
- "total_energy": Total energy.
Expand Down Expand Up @@ -743,8 +743,8 @@ class ParametersRunning(ParametersBase):
profiler_range : list
List with two entries determining with which batch/iteration number
the CUDA profiler will start and stop profiling. Please note that
this option only holds significance if the nsys profiler is used.
the CUDA profiler will start and stop profiling. Please note that
this option only holds significance if the nsys profiler is used.
"""

def __init__(self):
Expand Down

0 comments on commit ae2e0ed

Please sign in to comment.