diff --git a/docs/source/faqs.rst b/docs/source/faqs.rst index 2dce0c5..13ad4de 100644 --- a/docs/source/faqs.rst +++ b/docs/source/faqs.rst @@ -112,24 +112,25 @@ Model training .. _faq_epoch: -.. dropdown:: What is and how many epochs should I use for training? +.. dropdown:: How many epochs should I use for training? **What is an epoch?** An epoch refers to one complete pass through the entire training dataset. During an epoch, the model is trained on every sample in the dataset exactly once. Find more info - `here ` + `here `_ + (this link takes you to another set of docs specifically for Lightning Pose). + **With what value should I start?** To train a full model, we recommend starting with the default - 300. To get a baseline understanding of how the model performs, we recommend 50 epochs as the minimum number to get - a valid model to check. - **What are the trade-offs for increasing or decreasing that number?** - Increasing epochs may enhance convergence and accuracy but raise the risk of overfitting. + a valid model to check. + + **What are the trade-offs for increasing or decreasing the number of epochs?** + Increasing the epochs may enhance convergence and accuracy but raises the risk of overfitting. Conversely, fewer epochs might speed up training but risk underfitting. Balancing epochs is crucial to minimize both underfitting and overfitting. - - Post-processing --------------- diff --git a/docs/source/tabs/train_status.rst b/docs/source/tabs/train_status.rst index 52744f3..b9fb2c0 100644 --- a/docs/source/tabs/train_status.rst +++ b/docs/source/tabs/train_status.rst @@ -24,3 +24,35 @@ The earlier one is the supervised model, and the later one is the semi-supervise If you don't see all your models in that tab, hit the refresh button on the top right corner of the screen, and the other models should appear. + +Available metrics +----------------- + +The following are the important metrics for all model types +(supervised, context, semi-supervised, etc.): + +* ``train_supervised_loss``: this is the same as ``train_heatmap_mse_loss_weighted``, which is the + mean square error (MSE) between the true and predicted heatmaps on labeled training data +* ``train_supervised_rmse``: the root mean square error (RMSE) between the true and predicted + (x, y) coordinates on labeled training data; scale is in pixels +* ``val_supervised_loss``: this is the same as ``val_heatmap_mse_loss_weighted``, which is the + MSE between the true and predicted heatmaps on labeled validation data +* ``val_supervised_rmse``: the RMSE between the true and predicted (x, y) coordinates on labeled + validation data; scale is in pixels + +The following are important metrics for the semi-supervised models: + +* ``train_pca_multiview_loss_weighted``: the ``train_pca_multiview_loss`` (in pixels), which + measures multiview consistency, multplied by the loss weight set in the configuration file. + This metric is only computed on batches of unlabeled training data. +* ``train_pca_singleview_loss_weighted``: the ``train_pca_singleview_loss`` (in pixels), which + measures pose plausibility, multplied by the loss weight set in the configuration file. + This metric is only computed on batches of unlabeled training data. +* ``train_temporal_loss_weighted``: the ``train_temporal_loss`` (in pixels), which + measures temporal smoothness, multplied by the loss weight set in the configuration file. + This metric is only computed on batches of unlabeled training data. +* ``total_unsupervised_importance``: a weight on all *weighted* unsupervised losses that linearly + increases from 0 to 1 over 100 epochs +* ``total_loss``: weighted supervised loss (``train_heatmap_mse_loss_weighted``) plus + ``total_unsupervised_importance`` times the sum of all applicable weighted unsupervised losses + \ No newline at end of file