train.py

import torch\n\ndef train_model():\n    print('Training...')\n