Commit 8faaefc4 authored by buoyancy99's avatar buoyancy99 Committed by Francisco Massa
Browse files

update shallow copy method to allow variable size tensor input (#1111)

parent 95a87851
...@@ -30,7 +30,7 @@ class GeneralizedRCNNTransform(nn.Module): ...@@ -30,7 +30,7 @@ class GeneralizedRCNNTransform(nn.Module):
self.image_std = image_std self.image_std = image_std
def forward(self, images, targets=None): def forward(self, images, targets=None):
images = images[:] images = [img for img in images]
for i in range(len(images)): for i in range(len(images)):
image = images[i] image = images[i]
target = targets[i] if targets is not None else targets target = targets[i] if targets is not None else targets
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment