ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 10:42:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add .pt file extension as PyTorch

Browse Source 
Usually `.pt` is used as pytorch's extension, see [this StackOverflow question](https://stackoverflow.com/questions/59095824/what-is-difference-between-pt-pth-and-pwf-extentions-in-pytorch).

Furthermore `.pth` is used by Python to list additional package search paths (see [this PyTorch issue](https://github.com/pytorch/pytorch/issues/14864)) so IMO it might be worth reconsidering existence of it extension. AFAIK `.pt` is advised and used throughout most projects.
This commit is contained in:
Szymon Maszke 2020-01-06 13:08:54 +01:00 committed by GitHub
parent 54ae340ccb
commit 6815dd5410
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, ),
'.tflite': (tensorflowlite, ),
'.pth': (pytorch, ),
'.pt': (pytorch, ),
'.caffemodel': (caffe, ),
'.prototxt': (caffe, ),
'predict_net.pb': (caffe2, ),