Thank you!
So why do I get -
100.00% [30000/30000 01:58<00:00 Average Loss = inf]
Finished [100%]: Average Loss = nan
when I set X.shape[0] to get the Network up and running? And why does this produce a posterior and the trace?
And of course the most important - what can I do about it to get it fixed?
I also ask this because X.shape[1] does not seem to be a way that works.