Comment by codergautam

The code at the beginning was for training the 5M model I believe