Skip to content

Commit 9092535

Browse files
authored
Fix tuple unpacking bug (#19)
1 parent eb0a245 commit 9092535

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

treecrowndelineation/model/inference_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ def __init__(self, model):
88

99
def forward(self, x):
1010
output = self.model(x)
11-
if len(output) == 2:
11+
if isinstance(output, tuple) and len(output) == 2:
1212
y, metric = output
1313
y[:,:2] = torch.sigmoid(y[:,:2])
1414
return y, metric

0 commit comments

Comments
 (0)