diff --git a/trains/model.py b/trains/model.py
index a675d4e2..293ea154 100644
--- a/trains/model.py
+++ b/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, ),