ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-12 15:45:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #78 from szymonmaszke/patch-1

Browse Source 
Add .pt file extension as PyTorch
This commit is contained in:
Allegro AI 2020-01-06 17:30:31 +02:00 committed by GitHub
commit be8d100e33
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
trains/model.py
View File

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