121tester's picture
Refactor model files
cc78a1b verified
import torch\n\ndef train_model():\n print('Training...')\n