Added model.eval which was missing before evalution the model (sony#1180

)
reuvenperetz · Aug 21, 2024 · ed95fb2 · ed95fb2
1 parent d7fee57
commit ed95fb2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tutorials/notebooks/imx500_notebooks/pytorch/pytorch_yolov8n_for_imx500.ipynb b/tutorials/notebooks/imx500_notebooks/pytorch/pytorch_yolov8n_for_imx500.ipynb
@@ -132,7 +132,8 @@
    "source": [
     "from tutorials.mct_model_garden.models_pytorch.yolov8.yolov8 import ModelPyTorch, yaml_load, model_predict\n",
     "cfg_dict = yaml_load(\"tutorials/mct_model_garden/models_pytorch/yolov8/yolov8n.yaml\", append_filename=True)  # model dict\n",
-    "model = ModelPyTorch.from_pretrained(\"SSI-DNN/pytorch_yolov8n_detection_640x640\", cfg=cfg_dict)"
+    "model = ModelPyTorch.from_pretrained(\"SSI-DNN/pytorch_yolov8n_detection_640x640\", cfg=cfg_dict)\n",
+    "model.eval()"
    ],
    "metadata": {
     "collapsed": false