release commit

[jacinto-ai/pytorch-jacinto-ai-devkit.git] / modules / pytorch_jacinto_ai / vision / models / pixel2pixel / deeplabv3lite.py

diff --git a/modules/pytorch_jacinto_ai/vision/models/pixel2pixel/deeplabv3lite.py b/modules/pytorch_jacinto_ai/vision/models/pixel2pixel/deeplabv3lite.py
index b75366ea3cd213ece922141c7ff2be23193db6c8..2252e385ade719a7c839b8a3d3db75dbc2cb62f1 100644 (file)
--- a/modules/pytorch_jacinto_ai/vision/models/pixel2pixel/deeplabv3lite.py
+++ b/modules/pytorch_jacinto_ai/vision/models/pixel2pixel/deeplabv3lite.py
@@ -166,7 +166,7 @@ def deeplabv3lite_mobilenetv2_tv(model_config, pretrained=None):
     #
 
     if pretrained:
-        model = xnn.utils.load_weights_check(model, pretrained, change_names_dict, ignore_size=True, verbose=True)
+        model = xnn.utils.load_weights(model, pretrained, change_names_dict, ignore_size=True, verbose=True)
 
     return model, change_names_dict
 
@@ -192,7 +192,7 @@ def deeplabv3lite_mobilenetv2_ericsun(model_config, pretrained=None):
     #
 
     if pretrained:
-        model = xnn.utils.load_weights_check(model, pretrained, change_names_dict, ignore_size=True, verbose=True)
+        model = xnn.utils.load_weights(model, pretrained, change_names_dict, ignore_size=True, verbose=True)
 
     return model, change_names_dict
 
@@ -240,6 +240,6 @@ def deeplabv3lite_resnet50(model_config, pretrained=None):
     #
 
     if pretrained:
-        model = xnn.utils.load_weights_check(model, pretrained, change_names_dict, ignore_size=True, verbose=True)
+        model = xnn.utils.load_weights(model, pretrained, change_names_dict, ignore_size=True, verbose=True)
 
     return model, change_names_dict