I\'m trying to implement Adaptive cosine scaling on keras and train it on cloud TPU. I have tried a lot but am unable to fix the following error. I feel the error is either with
