add model

model_segvol_single.py

@@ -117,7 +117,11 @@ class SegVolModel(PreTrainedModel):
         logits_global_single[:, :, min_d:max_d+1, min_h:max_h+1, min_w:max_w+1] = logits_single_cropped
         return logits_global_single
 
-
+    def forward_train(self, image, train_organs, train_labels):
+        loss = self.model(image, text=None, boxes=None, points=None,
+                 train_organs=train_organs,
+                 train_labels=train_labels)
+        return loss
 
 # processor
 class SegVolProcessor():