Skip to content

Commit 1b3acc3

Browse files
committed
rename the pytorch extract model file.
1 parent b7af6de commit 1b3acc3

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

mmdnn/conversion/examples/pytorch/extract_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ def _main():
3232

3333
args = parser.parse_args()
3434

35-
file_name = "imagenet_{}.pt".format(args.network)
35+
file_name = "imagenet_{}.pth".format(args.network)
3636
if not os.path.exists(file_name):
3737
model = NETWORKS_MAP.get(args.network)
3838
model = model()

0 commit comments

Comments
 (0)