From ebb388cf4169874beaecb53d411a04a2baf45edb Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Wed, 12 Jul 2023 18:17:04 +0200
Subject: [PATCH] [models.pasa] Add comment on network output and loss
compatibility
---
src/ptbench/models/pasa.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/src/ptbench/models/pasa.py b/src/ptbench/models/pasa.py
index 5dd1c33c..7f0cf57d 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
----------
--
GitLab