i tried to train the model recently, but i found that some models required in the models/__init__.py are lost 
i tried to train the model recently, but i found that some models required in the models/init.py are lost
