diff --git a/src/ptbench/models/pasa.py b/src/ptbench/models/pasa.py index 5dd1c33c19c7a5b22e2b37bbfeb9943322f7f16f..7f0cf57dde2baf79cef18021543d6257a98e35a3 100644 --- a/src/ptbench/models/pasa.py +++ b/src/ptbench/models/pasa.py @@ -23,6 +23,9 @@ class Pasa(pl.LightningModule): Simple CNN for classification based on paper by [PASA-2019]_. + This network has a linear output. You should use losses with ``WithLogit`` + instead of cross-entropy versions when training. + Parameters ----------