We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 6cc76a7 commit caf6eabCopy full SHA for caf6eab
art/estimators/object_tracking/pytorch_goturn.py
@@ -241,7 +241,7 @@ def _get_losses(
241
242
for i in range(x.shape[0]):
243
x_i = inputs_t[i]
244
- y_pred = self.track(x=x_i, y_init=y_init[i])
+ y_pred = self._track(x=x_i, y_init=y_init[i])
245
gt_bb = labels_t[i]["boxes"]
246
loss = torch.nn.L1Loss(size_average=False)(y_pred.float(), gt_bb.float())
247
loss_list.append(loss)
0 commit comments