[translayer] Fix for fc without bias

Bug fix to work with fc layer with disabled bias.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>

diff --git a/test/input_gen/transLayer_v2.py b/test/input_gen/transLayer_v2.py
index db1dc36ed534e01088bab71076614c0d4473b1ca..c16035fc6a9fff095ea9ea3fb933bfa7536b2229 100644 (file)
--- a/test/input_gen/transLayer_v2.py
+++ b/test/input_gen/transLayer_v2.py
@@ -43,7 +43,10 @@ def fc_translate(model):
     params = [(name, tensor.detach()) for name, tensor in model.named_parameters()]
     def transpose_(weight):
         return (weight[0], weight[1].transpose(1, 0))
-    new_params = [transpose_(params[0]), params[1]]
+    if len(params) == 2:
+        new_params = [transpose_(params[0]), params[1]]
+    else:
+        new_params = [transpose_(params[0])]
     yield from new_params
 
 @register_for_(torch.nn.BatchNorm1d)