Merge branch 'dev' into main

README.md

@@ -2,7 +2,7 @@
 
 Disentanglement Library for pytorch and pytorch-lightning. With an easy to use configuration based on Hydra.
 
-
+**Documentation**: Check out the [docs and examples](https://docs.disent.dontpanic.sh/en/latest/)!
 
 ## Another disentanglement library?
 

requirements.txt

@@ -3,24 +3,24 @@ pip == 21.0
 # DATA SCIENCE & ML
 # =================
 numpy == 1.19.5
-pandas == 1.2.1
+# pandas == 1.2.1
 
-opencv-python == 4.5.1.48
+# opencv-python == 4.5.1.48
 
 torch == 1.7.1
 torchvision == 0.8.2
 torch-optimizer == 0.1.0
 pytorch-lightning == 1.1.5
-pytorch-lightning-bolts == 0.3.0
+# pytorch-lightning-bolts == 0.3.0
 
 # tensorflow == 2.4.1
 # tensorflow-gpu == 2.4.1
 # tensorboard == 2.4.1
 
 scipy == 1.6.0
 scikit-learn == 0.24.1
-xgboost == 1.3.3
-lightgbm == 3.1.1
+# xgboost == 1.3.3
+# lightgbm == 3.1.1
 
 # IMAGE AUGMENTATION
 # ==================
@@ -30,29 +30,29 @@ kornia == 0.4.1  # $ pip install git+https://github.com/kornia/kornia
 
 # INPUT / OUTPUT
 # ================
-psutil == 5.8.0
-pillow == 8.1.0
-imageio == 2.9.0
-moviepy == 1.0.3
-h5py == 2.10.0  # as of tensorflow 2.4 it does not support h5py 3+
+# psutil == 5.8.0
+# pillow == 8.1.0
+# imageio == 2.9.0
+# moviepy == 1.0.3
+h5py >= 2.10.0  # as of tensorflow 2.4 it does not support h5py 3+
 
 # GRAPHING & DISPLAY
 # ==================
 matplotlib == 3.3.3
-plotly == 4.14.3
-seaborn == 0.11.1
-tqdm == 4.56.0
+# plotly == 4.14.3
+# seaborn == 0.11.1
+# tqdm == 4.56.0
 
-streamlit == 0.75.0
-wandb == 0.10.14
+# streamlit == 0.75.0
+wandb == 0.10.15
 
 # UTILITY
 # =======
-pytest == 6.2.1
+# pytest == 6.2.1
 
 hydra-core == 1.0.5
 hydra-colorlog == 1.0.0
-hydra-joblib-launcher == 1.1.1
+# hydra-joblib-launcher == 1.1.1
 hydra-submitit-launcher == 1.0.1
 
-submitit == 1.1.5
+# submitit == 1.1.5