Open RichealYoung opened 5 years ago
Thanks for opening the first issue on our work! The parameter 'alpha' in PReluLayer should be a trainable tensor instead of float type,and the testing codes work well on our PC with tensorflow-gpu 1.10.0 and tensorlayer 1.11.1 Agree with @smarteric that the error is caused by the mismatching version of tensorlayer.
If you use tensorflow-gpu>=1.13 and tensorlayer 1.11.1, it will encounter this problem. Due to the mismatching version of tensorlayer. The easiest way to solve it is to write a PReluLayer function by yourself. Or just revise the activation.py in tensorlayer under your env. Using self.outputs = tf.maximum(alpha_var_constrained * self.inputs, self.inputs) instead of tf.nn.leaky_relu. Hope it helps
I ran the test.py and encountered the following error Error
python 3.6 tensorflow-gpu 1.13.1 tensorlayer 1.10.1