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