add device arg to load_model

Author: ueno-hiroshi002

tutorials/notebooks/task_notebooks/pytorch/example_deeplabv3p_pytorch_mixed_precision_ptq.ipynb

@@ -216,9 +216,10 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "device = torch.device(\"cuda:0\" if torch.cuda.is_available() else \"cpu\")\n",
     "model_builder = network.modeling.__dict__[\"deeplabv3plus_mobilenet\"]\n",
     "float_model = model_builder(output_stride=16)\n",
-    "float_model.load_state_dict(torch.load( \"best_deeplabv3plus_mobilenet_voc_os16.pth\", weights_only=False)['model_state'])"
+    "float_model.load_state_dict(torch.load( \"best_deeplabv3plus_mobilenet_voc_os16.pth\", weights_only=False, map_location=device)['model_state'])"
    ]
   },
   {
@@ -532,7 +533,7 @@
  ],
  "metadata": {
   "kernelspec": {
-   "display_name": "torch11 (3.10.12)",
+   "display_name": "py310-deeplabv3ptest (3.10.12)",
    "language": "python",
    "name": "python3"
   },