attmacah's picture
Create train.py
a9cebec verified