Skip to content
Snippets Groups Projects
Commit 917e86f3 authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

[doc] Add documentation about train-analysis

parent c726eab3
No related branches found
No related tags found
1 merge request!15Update documentation
......@@ -26,3 +26,4 @@ _work/
.mypy_cache/
.pytest_cache/
results*/
trainlog.pdf
......@@ -64,4 +64,23 @@ To train Pasa CNN on the Montgomery dataset:
mednet train -vv mlp shenzhen_rs --batch-size=4 --epochs=20
Plotting training metrics
-------------------------
Various metrics are recorded at each epoch during training, such as the execution time, loss and resource usage.
These are saved in a Tensorboard file, located in a `logs` subdirectory of the training output folder.
Mednet provides a :ref:`train-analysis <mednet.cli>` convenience script that graphs the scalars stored in these files and saves them in a pdf file.
Examples
========
Generates a pdf file with plots showing the evolution of logged metrics in time:
.. code:: sh
mednet train-analysis -vv <results/logs/folder> -o <results/trainlog.pdf>
.. include:: ../links.rst
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment