Skip to content
Snippets Groups Projects
Commit ebb388cf authored by André Anjos's avatar André Anjos :speech_balloon:
Browse files

[models.pasa] Add comment on network output and loss compatibility

parent f39ae93f
No related branches found
No related tags found
1 merge request!6Making use of LightningDataModule and simplification of data loading
...@@ -23,6 +23,9 @@ class Pasa(pl.LightningModule): ...@@ -23,6 +23,9 @@ class Pasa(pl.LightningModule):
Simple CNN for classification based on paper by [PASA-2019]_. 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 Parameters
---------- ----------
......
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