Chaitanya_C:

To train and experiment more, I’ve integrated the repository into my custom training pipeline by modularizing the code in such a way that the class has functions to build_net, train and eval the model. I have not made any change to the files for building the model, preprocessing or losses , except for a few functional changes .


I don’t know which changes were made, but you could try to isolate each change and check, if the change is applying the expected change.
Could you explain from a high-level perspective, what your changes were supposed to do?
Also, I assume you are using the softmax for debugging?