yuxin
commited on
Commit
·
da33dfe
1
Parent(s):
340c996
add model
Browse files- model_segvol_single.py +1 -0
model_segvol_single.py
CHANGED
|
@@ -118,6 +118,7 @@ class SegVolModel(PreTrainedModel):
|
|
| 118 |
return logits_global_single
|
| 119 |
|
| 120 |
def forward_train(self, image, train_organs, train_labels):
|
|
|
|
| 121 |
loss = self.model(image, text=None, boxes=None, points=None,
|
| 122 |
train_organs=train_organs,
|
| 123 |
train_labels=train_labels)
|
|
|
|
| 118 |
return logits_global_single
|
| 119 |
|
| 120 |
def forward_train(self, image, train_organs, train_labels):
|
| 121 |
+
print('in forward_train')
|
| 122 |
loss = self.model(image, text=None, boxes=None, points=None,
|
| 123 |
train_organs=train_organs,
|
| 124 |
train_labels=train_labels)
|