About training code

[MontaEllis]

Hi, I am on reproducing your training code, however I find the model is pretty small(2 MB). I wonder if it's my training data fault or what. My email : [email protected]