In that case you could try to overfit a small dataset, e.g. just 10 samples, and make sure your model is able to do so by playing around with hyperparameters.
If that’s still not working, there might be other issues I haven’t seen yet.