Update note for task3

AIM-Harvard · Nov 1, 2023 · e3e5700 · e3e5700
1 parent d1bdf82
commit e3e5700
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/experiments/supervised_training/foundation_finetune.yaml b/experiments/supervised_training/foundation_finetune.yaml
@@ -36,7 +36,7 @@ system:
     model:
         _target_: fmcib.ssl.models.LoadPretrainedResnet3D
         pretrained: "./models/ssl/fmcib.torch" 
-        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/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