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

[doc] Remove traces of signs-to-tb "model"

parent 7ecfd161
No related branches found
No related tags found
1 merge request!6Making use of LightningDataModule and simplification of data loading
......@@ -82,7 +82,7 @@ CNN and other models implemented.
ptbench.models.densenet
ptbench.models.normalizer
ptbench.models.logistic_regression
ptbench.models.signs_to_tb
ptbench.models.mlp
.. _ptbench.api.engines:
......
......@@ -26,8 +26,8 @@ Models
ptbench.models.config.densenet_pretrained
ptbench.models.config.densenet_rs
ptbench.models.config.logistic_regression
ptbench.models.config.mlp
ptbench.models.config.pasa
ptbench.models.config.signs_to_tb
.. _ptbench.config.datamodules:
......
......@@ -61,12 +61,13 @@ Montgomery dataset:
ptbench train -vv logistic_regression montgomery_rs --batch-size=4 --epochs=20
To train Signs_to_TB using predictions from DensenetForRS on the Shenzhen
To train an multi-layer perceptron (MLP) using predictions from a densenet
pre-trained to detect radiological findings (using NIH CXR-14), on the Shenzhen
dataset:
.. code:: sh
ptbench train -vv signs_to_tb shenzhen_rs --batch-size=4 --epochs=20
ptbench train -vv mlp shenzhen_rs --batch-size=4 --epochs=20
.. 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