Bhaveshkumar Dharmani
Results
1
comments of
Bhaveshkumar Dharmani
The problem still persist for both GPU and TPU. Even a simple sample-code shows the same error. x = np.random.rand(1, 10, 10, 1) dense = addon_layers.WeightNormalization(tf.keras.layers.Dense(10), data_init=False) y = dense(x)...