frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update default ouput extension in optimize_for_mobile.cc (#45598)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/45598

.bc is causing issues on Android.  Let's switch to .ptl.

Test Plan: CI

Reviewed By: kimishpatel

Differential Revision: D24026180

fbshipit-source-id: 9f252f3652d748bccb19dc61a783d693e171b2c6
This commit is contained in:
David Reiss
2020-10-15 15:31:53 -07:00
committed by Facebook GitHub Bot
parent f96cb9de79
commit faf03bd226
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
binaries/optimize_for_mobile.cc
View File

@ -51,7 +51,7 @@ int main(int argc, char** argv) {
CAFFE_ENFORCE(FLAGS_model != "", c10::UsageMessage());
std::string output_model_name =
FLAGS_model.substr(0, FLAGS_model.find(".")) + "_optimized.bc";
FLAGS_model.substr(0, FLAGS_model.find(".")) + "_optimized.ptl";
if (FLAGS_output != "") {
output_model_name = FLAGS_output;