Skip to content

Commit

Permalink
check
Browse files Browse the repository at this point in the history
  • Loading branch information
ydshieh committed Oct 17, 2024
1 parent b8d3323 commit cc77eba
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -828,7 +828,7 @@ def test_grounding_dino_loss(self):
"loss_giou_enc": torch.tensor(0.7380),
}

expected_loss = torch.tensor(32481.0371)
expected_loss = torch.tensor(32482.2305)

for key in expected_loss_dict:
self.assertTrue(torch.allclose(outputs.loss_dict[key], expected_loss_dict[key], atol=1e-3))
Expand Down

0 comments on commit cc77eba

Please sign in to comment.