Upload model

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:90723bf05df5c63f9b9b783940265ed993dd78c0277f643cca8d8b442a14051a
+oid sha256:17fda23b776cda909f3f20772fa4d4bc8bd677ba2ad2bbc469b8f74eddea8365
 size 342867728

modeling_resnet3d.py

@@ -39,4 +39,4 @@ class Resnet3DScrollprizeModel(PreTrainedModel):
         feat_maps = self.backbone(tensor)
         feat_maps_pooled = [torch.max(f, dim=2)[0] for f in feat_maps]
         pred_mask = self.decoder(feat_maps_pooled)
-        return x
+        return pred_mask