We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 36ccfb1 commit 909727eCopy full SHA for 909727e
tests/unit/algorithms/detection/adapters/mmdet/models/heads/test_custom_dino_head.py
@@ -85,10 +85,10 @@ def setup(self):
85
@e2e_pytest_unit
86
def test_forward_train(self):
87
inputs = [
88
- torch.randn([2, 256, 92, 95]),
89
- torch.randn([2, 256, 46, 48]),
90
- torch.randn([2, 256, 23, 24]),
91
- torch.randn([2, 256, 12, 12]),
+ torch.zeros([2, 256, 92, 95]),
+ torch.zeros([2, 256, 46, 48]),
+ torch.zeros([2, 256, 23, 24]),
+ torch.zeros([2, 256, 12, 12]),
92
]
93
gt_bboxes = [
94
torch.Tensor(
@@ -171,10 +171,10 @@ def test_forward_train(self):
171
172
def test_simple_test_bboxes(self):
173
feats = [
174
- torch.randn([2, 256, 100, 134]),
175
- torch.randn([2, 256, 50, 67]),
176
- torch.randn([2, 256, 25, 34]),
177
- torch.randn([2, 256, 13, 17]),
+ torch.zeros([2, 256, 100, 134]),
+ torch.zeros([2, 256, 50, 67]),
+ torch.zeros([2, 256, 25, 34]),
+ torch.zeros([2, 256, 13, 17]),
178
179
img_metas = [
180
{
0 commit comments