Merge branch 'master' into rev_1

AIM-Harvard · Dec 8, 2023 · 2fef9db · 2fef9db
2 parents 1abf168 + c524087
commit 2fef9db
Show file tree

Hide file tree

Showing 6 changed files with 16 additions and 38 deletions.
diff --git a/.github/dependabot.yml b/.github/dependabot.yml
diff --git a/experiments/supervised_training/foundation_finetune.yaml b/experiments/supervised_training/foundation_finetune.yaml
@@ -43,6 +43,7 @@ system:
         weights_path: "./fmcib.torch" 
         heads:  [4096, 2048, 512, 256, 1] # Note: Change to [4096, 2048, 1] for Task 2 and [4096, Task 3
 
+
     post_criterion_activation: "$torch.sigmoid" # Note: Change to $torch.sigmoid for Task 2 and Task 3
     cast_target_dtype_to: "$torch.float32" # Note: Uncomment for Task 2 and Task 3
 

diff --git a/experiments/supervised_training/supervised_finetune.yaml b/experiments/supervised_training/supervised_finetune.yaml
@@ -36,7 +36,7 @@ system:
     model:
         _target_: fmcib.ssl.models.LoadPretrainedResnet3D
         pretrained: "./models/supervised/task1_supervised.torch" 
-        heads: [4096, 2048, 1] 
+        heads: [4096, 2048, 1]   # Note: Change to [4096, 2048, 512, 256, 1] for Task 3
 
     post_criterion_activation: "$torch.sigmoid" 
     cast_target_dtype_to: "$torch.float32"

diff --git a/experiments/supervised_training/supervised_random_init.yaml b/experiments/supervised_training/supervised_random_init.yaml
@@ -35,7 +35,7 @@ system:
     num_workers: 3
     model:
         _target_: fmcib.ssl.models.LoadPretrainedResnet3D
-        heads: [4096, 2048, 8] # Note: Change to [4096, 2048, 1] for Task 2 and Task 3
+        heads: [4096, 2048, 8] # Note: Change to [4096, 2048, 1] for Task 2 and [4096, 2048, 512, 256, 1] for Task 3
 
     post_criterion_activation: "$torch.nn.functional.softmax" # Note: Change to $torch.sigmoid for Task 2 and Task 3
     # cast_target_dtype_to: "$torch.float32" # Note: Uncomment for Task 2 and Task 3

diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -51,6 +51,7 @@ aiohttp = "^3.8.6"
 fonttools = "^4.44.1"
 pydantic = "^2.5.0"
 
+
 [tool.poetry.group.dev.dependencies]
 black = "^23.7.0"
 bandit = "^1.7.1"