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):
self.image_std = image_std
def forward(self, images, targets=None):
images = images[:]
images = [img for img in images]
for i in range(len(images)):
image = images[i]
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