How to make use of class_weights to calculated custom loss fuction while using custom training loop (i.e. not using .fit )

Question

I have written my custom training loop using tf.GradientTape(). My data has 2 classes. The classes are not balanced; class1 data contributes almost 80% and class2 contribute