bug fixes

src/test/resources/target_code/gluon/ganModel/defaultGAN/CNNCreator_defaultGAN_defaultGANConnector_predictor.py

@@ -170,6 +170,9 @@ class CNNCreator_defaultGAN_defaultGANConnector_predictor:
         for i, network in self.networks.items():
             network.export(self._model_dir_ + self._model_prefix_ + "_" + str(i), epoch=0)
 
+    def setWeightInitializer(self, initializer):
+        self.weight_initializer = initializer
+
     def getInputs(self):
         inputs = {}
         input_dimensions = (100,)

src/test/resources/target_code/gluon/ganModel/infoGAN/CNNCreator_infoGAN_infoGANConnector_predictor.py

@@ -170,6 +170,9 @@ class CNNCreator_infoGAN_infoGANConnector_predictor:
         for i, network in self.networks.items():
             network.export(self._model_dir_ + self._model_prefix_ + "_" + str(i), epoch=0)
 
+    def setWeightInitializer(self, initializer):
+        self.weight_initializer = initializer
+
     def getInputs(self):
         inputs = {}
         input_dimensions = (62,)

src/test/resources/target_code/gluon/ganModel/infoGAN/gan/CNNCreator_infoGAN_infoGANQNetwork.py

@@ -170,6 +170,9 @@ class CNNCreator_infoGAN_infoGANQNetwork:
         for i, network in self.networks.items():
             network.export(self._model_dir_ + self._model_prefix_ + "_" + str(i), epoch=0)
 
+    def setWeightInitializer(self, initializer):
+        self.weight_initializer = initializer
+
     def getInputs(self):
         inputs = {}
         input_dimensions = (512,4,4,)

src/test/resources/target_code/gluon/reinforcementModel/cartpole/CNNTrainer_cartpole_master_dqn.py

@@ -46,7 +46,9 @@ if __name__ == "__main__":
     env = reinforcement_learning.environment.GymEnvironment('CartPole-v0')
 
     context = mx.cpu()
+    initializer = mx.init.Normal()
     qnet_creator = CNNCreator_cartpole_master_dqn.CNNCreator_cartpole_master_dqn()
+    qnet_creator.setWeightInitializer(initializer)
     qnet_creator.construct(context)
 
     agent_params = {

src/test/resources/target_code/gluon/reinforcementModel/mountaincar/CNNTrainer_mountaincar_master_actor.py

@@ -47,9 +47,13 @@ if __name__ == "__main__":
     env = reinforcement_learning.environment.GymEnvironment('MountainCarContinuous-v0')
 
     context = mx.cpu()
+    initializer = mx.init.Normal()
+    critic_initializer = mx.init.Normal()
     actor_creator = CNNCreator_mountaincar_master_actor.CNNCreator_mountaincar_master_actor()
+    actor_creator.setWeightInitializer(initializer)
     actor_creator.construct(context)
     critic_creator = CNNCreator_mountaincar_agent_mountaincarCritic()
+    critic_creator.setWeightInitializer(critic_initializer)
     critic_creator.construct(context)
 
     agent_params = {

src/test/resources/target_code/gluon/reinforcementModel/mountaincar/reinforcement_learning/CNNCreator_mountaincar_agent_mountaincarCritic.py

@@ -170,6 +170,9 @@ class CNNCreator_mountaincar_agent_mountaincarCritic:
         for i, network in self.networks.items():
             network.export(self._model_dir_ + self._model_prefix_ + "_" + str(i), epoch=0)
 
+    def setWeightInitializer(self, initializer):
+        self.weight_initializer = initializer
+
     def getInputs(self):
         inputs = {}
         input_dimensions = (2,)

src/test/resources/target_code/gluon/reinforcementModel/torcs/CNNTrainer_torcs_agent_torcsAgent_dqn.py

@@ -53,7 +53,9 @@ if __name__ == "__main__":
     env = reinforcement_learning.environment.RosEnvironment(**env_params)
 
     context = mx.cpu()
+    initializer = mx.init.Normal()
     qnet_creator = CNNCreator_torcs_agent_torcsAgent_dqn.CNNCreator_torcs_agent_torcsAgent_dqn()
+    qnet_creator.setWeightInitializer(initializer)
     qnet_creator.construct(context)
 
     agent_params = {

src/test/resources/target_code/gluon/reinforcementModel/torcs_td3/CNNTrainer_torcs_agent_torcsAgent_actor.py

@@ -54,9 +54,13 @@ if __name__ == "__main__":
     env = reinforcement_learning.environment.RosEnvironment(**env_params)
 
     context = mx.gpu()
+    initializer = mx.init.Normal()
+    critic_initializer = mx.init.Normal()
     actor_creator = CNNCreator_torcs_agent_torcsAgent_actor.CNNCreator_torcs_agent_torcsAgent_actor()
+    actor_creator.setWeightInitializer(initializer)
     actor_creator.construct(context)
     critic_creator = CNNCreator_torcs_agent_network_torcsCritic()
+    critic_creator.setWeightInitializer(critic_initializer)
     critic_creator.construct(context)
 
     agent_params = {

src/test/resources/target_code/gluon/reinforcementModel/torcs_td3/reinforcement_learning/CNNCreator_torcs_agent_network_torcsCritic.py

@@ -170,6 +170,9 @@ class CNNCreator_torcs_agent_network_torcsCritic:
         for i, network in self.networks.items():
             network.export(self._model_dir_ + self._model_prefix_ + "_" + str(i), epoch=0)
 
+    def setWeightInitializer(self, initializer):
+        self.weight_initializer = initializer
+
     def getInputs(self):
         inputs = {}
         input_dimensions = (29,)