We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent f14238f commit cdb299cCopy full SHA for cdb299c
modules/model_definitions/README.md
@@ -0,0 +1,8 @@
1
+# Model Definition Module
2
+
3
+This module contains the code for all models used in this work.
4
+> ```def_net``` contains the model definition for the small MLP or CNN models in our own model zoos.
5
+> ```def_simclr_ae_module``` contains the training class for our neural representation learning models.
6
+> ```def_simclr_ae_trainable``` contains the ray.tune trainable wrapper around the module
7
+> ```./components/``` contains encoder and decoder architectures, as well as the different encodings and the loss definitions.
8
+> ```./downstream_tasks/``` contains a wrapper class around the downstream tasks as well as the baseline models.
0 commit comments