Upload model
Browse files- model.safetensors +1 -1
- modeling_swinunetrv2.py +1 -1
model.safetensors
CHANGED
|
@@ -1,3 +1,3 @@
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
-
oid sha256:
|
| 3 |
size 298598428
|
|
|
|
| 1 |
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:e8ec15b5256e65c44e2b147c04db7c0046ee9d4f3a4c37f118e3454320b63ee3
|
| 3 |
size 298598428
|
modeling_swinunetrv2.py
CHANGED
|
@@ -79,7 +79,7 @@ class SwinUNETRv2ForImageSegmentation(SwinUNETRv2PreTrainedModel):
|
|
| 79 |
use_v2=True,
|
| 80 |
)
|
| 81 |
|
| 82 |
-
def forward(self, tensor, train=
|
| 83 |
|
| 84 |
criterion = DiceCELoss(to_onehot_y=True, softmax=True)
|
| 85 |
|
|
|
|
| 79 |
use_v2=True,
|
| 80 |
)
|
| 81 |
|
| 82 |
+
def forward(self, tensor, train=False, roi_size=(128, 128, 128), sw_batch_size=1):
|
| 83 |
|
| 84 |
criterion = DiceCELoss(to_onehot_y=True, softmax=True)
|
| 85 |
|