Output is returnes as a tuple rather than a torch.size([])

madhavkhoslaa · Jul 15, 2019 · d0784f2 · d0784f2
1 parent 630fa0a
commit d0784f2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils.py b/utils.py
@@ -46,5 +46,5 @@ def get_out_size(model):
                 outputs = model(
                     inputs.type(
                         torch.FloatTensor))
-            return outputs.size()
+            return tuple(outputs.size())