Merge pull request #78 from szymonmaszke/patch-1

Add .pt file extension as PyTorch
2025-02-07 21:33:25 +00:00 · 2020-01-06 17:30:31 +02:00 · 2020-01-06 17:30:31 +02:00 · be8d100e33
commit be8d100e33
parent 30eaed79ea 6815dd5410
1 changed files with 1 additions and 0 deletions
--- 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, ),