Neuroschemata

I suspect the source of the problem is unrelated to the model size. With the default parameters using the command you posted above, I get the following: |batch size |...

Please post the exact command you used to train the model so that we can help diagnose.