Merge pull request #61 from husthyc/master

[iOS] Update the Hello World example using Lite interpreter

Author: xta0

HelloWorld/HelloWorld/HelloWorld/TorchBridge/TorchModule.mm

@@ -1,17 +1,16 @@
 #import "TorchModule.h"
-#import <LibTorch/LibTorch.h>
+#import <Libtorch-Lite/Libtorch-Lite.h>
 
 @implementation TorchModule {
  @protected
-  torch::jit::script::Module _impl;
+  torch::jit::mobile::Module _impl;
 }
 
 - (nullable instancetype)initWithFileAtPath:(NSString*)filePath {
   self = [super init];
   if (self) {
     try {
-      _impl = torch::jit::load(filePath.UTF8String);
-      _impl.eval();
+      _impl = torch::jit::_load_for_mobile(filePath.UTF8String);
     } catch (const std::exception& exception) {
       NSLog(@"%s", exception.what());
       return nil;

HelloWorld/HelloWorld/HelloWorld/model/model.pt

@@ -1,5 +1,5 @@
 
 platform :ios, '12.0'
 target 'HelloWorld' do
-    pod 'LibTorch', '~> 1.9.0'
+    pod 'LibTorch-Lite', '~> 1.9.0'
 end

HelloWorld/HelloWorld/Podfile

@@ -7,4 +7,4 @@
 example = torch.rand(1, 3, 224, 224)
 traced_script_module = torch.jit.trace(model, example)
 torchscript_model_optimized = optimize_for_mobile(traced_script_module)
-torchscript_model_optimized.save("HelloWorld/HelloWorld/model/model.pt")
+torchscript_model_optimized._save_for_lite_interpreter("HelloWorld/HelloWorld/model/model.pt")