Add TODO for setting weights_only to True

https://pytorch.org/docs/stable/generated/torch.load.html

Author: huydhn

examples/qualcomm/oss_scripts/ssd300_vgg16.py

@@ -109,6 +109,8 @@ def SSD300VGG16(pretrained_weight_model):
     from model import SSD300
 
     model = SSD300(n_classes=21)
+    # TODO: If possible, it's better to set weights_only to True
+    # https://pytorch.org/docs/stable/generated/torch.load.html
     checkpoint = torch.load(
         pretrained_weight_model, map_location="cpu", weights_only=False
     )

examples/qualcomm/scripts/mobilebert_fine_tune.py

@@ -204,6 +204,8 @@ def get_fine_tuned_mobilebert(artifacts_dir, pretrained_weight, batch_size):
             )
 
     model.load_state_dict(
+        # TODO: If possible, it's better to set weights_only to True
+        # https://pytorch.org/docs/stable/generated/torch.load.html
         torch.load(
             (
                 f"{artifacts_dir}/finetuned_mobilebert_epoch_{epochs}.model"

exir/serde/export_serialize.py

@@ -242,6 +242,8 @@ def deserialize_torch_artifact(serialized: bytes):
         return {}
     buffer = io.BytesIO(serialized)
     buffer.seek(0)
+    # TODO: If possible, it's better to set weights_only to True
+    # https://pytorch.org/docs/stable/generated/torch.load.html
     return torch.load(buffer, weights_only=False)