enable flop printing & logging at the beginning of train & test
@@ -163,6 +163,7 @@ class Instances:
return instance_lists[0]
image_size = instance_lists[0].image_size
+ if not isinstance(image_size, torch.Tensor): # could be a tensor in tracing
for i in instance_lists[1:]:
assert i.image_size == image_size
ret = Instances(image_size)
