I am planning to add class weights to give extra weightage for my classes which has lesser count. I got the concept but I am stuck with where to add this. Can I add this in pipeline config file?? If yes where?.

I am training EfficienDet D4 on my custom dataset using TensorFlow Object Detection API, I have an imbalanced dataset so, I am planning to add class weights to give extra weightage for my classes which has lesser count. I got the concept but I am stuck with where to add this. Can I add this in pipeline config file?? If yes where?. Otherwise which is the file where model.fit lies. Please give suggestions. Or if there is better way of doing this, please let me know. where does class_weights or weighted loss penalize the network? Actually I want to implement the solutions from the above link

tensorflow / models

I am planning to add class weights to give extra weightage for my classes which has lesser count. I got the concept but I am stuck with where to add this. Can I add this in pipeline config file?? If yes where?. #10533