Env: Tensorflow2.3.0 python3.6
I\'m trying to custom a layer for training process for image augmentation. Here\'s my code:
class RandomLight(layers.Layer):
