fix(torch/native): prevent loading weights before instanciating nativ…

…e model
jolibrain · Oct 21, 2020 · b15d767 · b15d767
1 parent 77a016b
commit b15d767
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/backends/torch/torchlib.cc b/src/backends/torch/torchlib.cc
@@ -196,7 +196,8 @@ namespace dd
 
   void TorchModule::native_model_load(const TorchModel &tmodel)
   {
-    if (!tmodel._native.empty())
+    if (!tmodel._native.empty()
+        && _native != nullptr) //_native has to be instanciated before loading
       {
         _logger->info("loading " + tmodel._native);
         try