Number of parameters in model summary is 0 for my custom model

[Harshit_Pandey]

I do get grads for every iteration but they are very small it seems.