thanks for the reply. My model’s is recurring over a layer instead of a rnn/lstm cell, so that explains the abnormal number of parameters and even my batches are taking 15s!!
Also, I see my training loss fluctuating…it is stuck at some value is fluctuating around it after around 200 batches…