Merge pull request #7 from pme0/registry

Registry
pme0 · Nov 12, 2023 · 1eec314 · 1eec314
2 parents 39a209b + 8f39077
commit 1eec314
Show file tree

Hide file tree

Showing 62 changed files with 745 additions and 1,199 deletions.
diff --git a/.gitignore b/.gitignore
@@ -9,6 +9,7 @@ tests/mlruns/
 tests/logs/
 tests/wandb/
 tests/lightning_logs/
+# hydra
 outputs/
 
 # model weights

diff --git a/conda_env.yaml b/conda_env.yaml
@@ -3,8 +3,8 @@ channels:
   - defaults
   - conda-forge
 dependencies:
-  - pip==23.0.1
-  - python==3.10 # numpy==1.22 not working with python==3.9
+  - pip
+  - python==3.10
   - pip:
     - colorama==0.4.4
     - deepspeed==0.5.10
@@ -17,13 +17,14 @@ dependencies:
     - matplotlib==3.5.1
     - numpy==1.23.5
     - omegaconf==2.1.1
-    - opencv-python==4.7.0.72 #==4.1.2.30
+    - opencv-python==4.7.0.72
     - pandas==1.5.3
+    - pytest
     - pyyaml==6.0
     - seaborn==0.12.0
-    - torch==2.0.0 #==1.13.1
+    - torch==2.0.0
     - torchaudio==2.0.1
     - torchlibrosa==0.1.0
-    - torchmetrics==0.11.4 #0.11.0
-    - torchvision==0.15.1  #0.11.3
+    - torchmetrics==1.2.0
+    - torchvision==0.15.1
     - wandb==0.12.21
diff --git a/configs/AudioClassification_lstm.yaml b/configs/AudioClassification_lstm.yaml
diff --git a/configs/ImageClassification_cnn.yaml b/configs/ImageClassification_cnn.yaml
diff --git a/configs/ImageClassification_cnn_TaskAgnostic.yaml b/configs/ImageClassification_cnn_TaskAgnostic.yaml
diff --git a/configs/ImageClassification_resnet.yaml b/configs/ImageClassification_resnet.yaml
diff --git a/configs/ImageGeneration_gan.yaml b/configs/ImageGeneration_gan.yaml