Skip to content

Commit cdb299c

Browse files
authored
create README for model_definitions module
1 parent f14238f commit cdb299c

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

modules/model_definitions/README.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)