aapot's picture
Add 300k train step pytorch model
3212143