remove deprecated pos_embed

Signed-off-by: YunLiu <[email protected]>
Project-MONAI · Sep 10, 2024 · 486c8cd · 486c8cd
1 parent ae7496e
commit 486c8cd
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 6 deletions.
diff --git a/3d_segmentation/unetr_btcv_segmentation_3d.ipynb b/3d_segmentation/unetr_btcv_segmentation_3d.ipynb
@@ -586,7 +586,7 @@
     "    hidden_size=768,\n",
     "    mlp_dim=3072,\n",
     "    num_heads=12,\n",
-    "    pos_embed=\"perceptron\",\n",
+    "    proj_type=\"perceptron\",\n",
     "    norm_name=\"instance\",\n",
     "    res_block=True,\n",
     "    dropout_rate=0.0,\n",

diff --git a/3d_segmentation/unetr_btcv_segmentation_3d_lightning.ipynb b/3d_segmentation/unetr_btcv_segmentation_3d_lightning.ipynb
@@ -423,7 +423,7 @@
     "            hidden_size=768,\n",
     "            mlp_dim=3072,\n",
     "            num_heads=12,\n",
-    "            pos_embed=\"perceptron\",\n",
+    "            proj_type=\"perceptron\",\n",
     "            norm_name=\"instance\",\n",
     "            res_block=True,\n",
     "            conv_block=True,\n",

diff --git a/deepedit/ignite/train.py b/deepedit/ignite/train.py
@@ -76,7 +76,7 @@ def get_network(network, labels, spatial_size):
             hidden_size=1536,
             mlp_dim=3072,
             num_heads=48,
-            pos_embed="conv",
+            proj_type="conv",
             norm_name="instance",
             res_block=True,
         )

diff --git a/self_supervised_pretraining/vit_unetr_ssl/multi_gpu/mgpu_ssl_train.py b/self_supervised_pretraining/vit_unetr_ssl/multi_gpu/mgpu_ssl_train.py
@@ -175,7 +175,7 @@ def main(args):
         in_channels=1,
         img_size=(96, 96, 96),
         patch_size=(16, 16, 16),
-        pos_embed="conv",
+        proj_type="conv",
         hidden_size=768,
         mlp_dim=3072,
     )

diff --git a/self_supervised_pretraining/vit_unetr_ssl/ssl_finetune.ipynb b/self_supervised_pretraining/vit_unetr_ssl/ssl_finetune.ipynb
@@ -317,7 +317,7 @@
     "    hidden_size=768,\n",
     "    mlp_dim=3072,\n",
     "    num_heads=12,\n",
-    "    pos_embed=\"conv\",\n",
+    "    proj_type=\"conv\",\n",
     "    norm_name=\"instance\",\n",
     "    res_block=True,\n",
     "    dropout_rate=0.0,\n",

diff --git a/self_supervised_pretraining/vit_unetr_ssl/ssl_train.ipynb b/self_supervised_pretraining/vit_unetr_ssl/ssl_train.ipynb
@@ -253,7 +253,7 @@
     "    in_channels=1,\n",
     "    img_size=(96, 96, 96),\n",
     "    patch_size=(16, 16, 16),\n",
-    "    pos_embed=\"conv\",\n",
+    "    proj_type=\"conv\",\n",
     "    hidden_size=768,\n",
     "    mlp_dim=3072,\n",
     ")\n",