mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
89c4e8c22b
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/67746 Test Plan: Visual inspection. Sandcastle. Reviewed By: zertosh Differential Revision: D31986646 fbshipit-source-id: 91885c20c3cead3853c49abb9fe0a94a67f33cc8
21 lines
450 B
C++
21 lines
450 B
C++
#include <torch/csrc/jit/api/module.h>
|
|
#include <torch/jit.h>
|
|
#include <torch/script.h>
|
|
|
|
#include <fstream>
|
|
#include <iostream>
|
|
#include <string>
|
|
|
|
int main(int argc, char* argv[]) {
|
|
std::string input_file_path{argv[1]};
|
|
std::string output_file_path{argv[2]};
|
|
|
|
std::ifstream ifs(input_file_path);
|
|
std::stringstream buffer;
|
|
buffer << ifs.rdbuf();
|
|
torch::jit::Module m("TestModule");
|
|
|
|
m.define(buffer.str());
|
|
m.save(output_file_path);
|
|
}
|