diff --git a/src/main/resources/templates/gluon/pythonExecute.ftl b/src/main/resources/templates/gluon/pythonExecute.ftl index c0093eeb201949a7a820d44136ecfc073fa1d645..1f2dcb306dffb3bbe8c3c3217d3e12487c223dcf 100644 --- a/src/main/resources/templates/gluon/pythonExecute.ftl +++ b/src/main/resources/templates/gluon/pythonExecute.ftl @@ -2,7 +2,7 @@ ${member} = mx.nd.zeroes((${tc.join(tc.getLayerVariableMembers("batch_size")[member], ", ")},), ctx=mx_context) </#list> <#list tc.architecture.outputs as output> - ${tc.getName(output)} = mx.nd.zeroes(((${tc.join(output.ioDeclaration.type.dimensions, ", ")},), ctx=mx_context) + ${tc.getName(output)} = mx.nd.zeroes((${tc.join(output.ioDeclaration.type.dimensions, ", ")},), ctx=mx_context) </#list> <#list tc.architecture.streams as stream> diff --git a/src/test/resources/target_code/CNNSupervisedTrainer_Alexnet.py b/src/test/resources/target_code/CNNSupervisedTrainer_Alexnet.py index 0d22884f583aabaa3969e147bf798ae38daa1ed8..51b7976921f22d15b1bd108b720ed6eec021e17b 100644 --- a/src/test/resources/target_code/CNNSupervisedTrainer_Alexnet.py +++ b/src/test/resources/target_code/CNNSupervisedTrainer_Alexnet.py @@ -136,7 +136,7 @@ class CNNSupervisedTrainer_Alexnet: predictions_label = batch.label[0].as_in_context(mx_context) with autograd.record(): - predictions_ = mx.nd.zeroes(((10,), ctx=mx_context) + predictions_ = mx.nd.zeroes((10,), ctx=mx_context) predictions_ = self._networks[0](data_) @@ -173,7 +173,7 @@ class CNNSupervisedTrainer_Alexnet: ] if True: - predictions_ = mx.nd.zeroes(((10,), ctx=mx_context) + predictions_ = mx.nd.zeroes((10,), ctx=mx_context) predictions_ = self._networks[0](data_) @@ -194,7 +194,7 @@ class CNNSupervisedTrainer_Alexnet: ] if True: - predictions_ = mx.nd.zeroes(((10,), ctx=mx_context) + predictions_ = mx.nd.zeroes((10,), ctx=mx_context) predictions_ = self._networks[0](data_) diff --git a/src/test/resources/target_code/CNNSupervisedTrainer_CifarClassifierNetwork.py b/src/test/resources/target_code/CNNSupervisedTrainer_CifarClassifierNetwork.py index 2a8aee064f211ae02c784dcffdc32d9089f273e3..bf5ee75f1f2d16bed187f8bf1a5ffcdecb692a46 100644 --- a/src/test/resources/target_code/CNNSupervisedTrainer_CifarClassifierNetwork.py +++ b/src/test/resources/target_code/CNNSupervisedTrainer_CifarClassifierNetwork.py @@ -136,7 +136,7 @@ class CNNSupervisedTrainer_CifarClassifierNetwork: softmax_label = batch.label[0].as_in_context(mx_context) with autograd.record(): - softmax_ = mx.nd.zeroes(((10,), ctx=mx_context) + softmax_ = mx.nd.zeroes((10,), ctx=mx_context) softmax_ = self._networks[0](data_) @@ -173,7 +173,7 @@ class CNNSupervisedTrainer_CifarClassifierNetwork: ] if True: - softmax_ = mx.nd.zeroes(((10,), ctx=mx_context) + softmax_ = mx.nd.zeroes((10,), ctx=mx_context) softmax_ = self._networks[0](data_) @@ -194,7 +194,7 @@ class CNNSupervisedTrainer_CifarClassifierNetwork: ] if True: - softmax_ = mx.nd.zeroes(((10,), ctx=mx_context) + softmax_ = mx.nd.zeroes((10,), ctx=mx_context) softmax_ = self._networks[0](data_) diff --git a/src/test/resources/target_code/CNNSupervisedTrainer_VGG16.py b/src/test/resources/target_code/CNNSupervisedTrainer_VGG16.py index 7651ff559662231a4edd5be811f407c81008decc..d98fe840cd4221780ec76063c9cd3067b8b2ccbd 100644 --- a/src/test/resources/target_code/CNNSupervisedTrainer_VGG16.py +++ b/src/test/resources/target_code/CNNSupervisedTrainer_VGG16.py @@ -136,7 +136,7 @@ class CNNSupervisedTrainer_VGG16: predictions_label = batch.label[0].as_in_context(mx_context) with autograd.record(): - predictions_ = mx.nd.zeroes(((1000,), ctx=mx_context) + predictions_ = mx.nd.zeroes((1000,), ctx=mx_context) predictions_ = self._networks[0](data_) @@ -173,7 +173,7 @@ class CNNSupervisedTrainer_VGG16: ] if True: - predictions_ = mx.nd.zeroes(((1000,), ctx=mx_context) + predictions_ = mx.nd.zeroes((1000,), ctx=mx_context) predictions_ = self._networks[0](data_) @@ -194,7 +194,7 @@ class CNNSupervisedTrainer_VGG16: ] if True: - predictions_ = mx.nd.zeroes(((1000,), ctx=mx_context) + predictions_ = mx.nd.zeroes((1000,), ctx=mx_context) predictions_ = self._networks[0](data_)