We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent eb0a245 commit 9092535Copy full SHA for 9092535
1 file changed
treecrowndelineation/model/inference_model.py
@@ -8,7 +8,7 @@ def __init__(self, model):
8
9
def forward(self, x):
10
output = self.model(x)
11
- if len(output) == 2:
+ if isinstance(output, tuple) and len(output) == 2:
12
y, metric = output
13
y[:,:2] = torch.sigmoid(y[:,:2])
14
return y, metric
0 commit comments