Merge pull request #78 from szymonmaszke/patch-1

Add .pt file extension as PyTorch

trains/model.py

@@ -53,6 +53,7 @@ class Framework(Options):
         'model.json': (tensorflowjs, ),
         '.tflite': (tensorflowlite, ),
         '.pth': (pytorch, ),
+        '.pt': (pytorch, ),
         '.caffemodel': (caffe, ),
         '.prototxt': (caffe, ),
         'predict_net.pb': (caffe2, ),