fix criterion weight load error in notebooks

drprojects · Aug 21, 2023 · fd0ba28 · fd0ba28
1 parent 423f154
commit fd0ba28
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 8 deletions.
diff --git a/notebooks/demo_dales.ipynb b/notebooks/demo_dales.ipynb
@@ -85,8 +85,7 @@
     "model = hydra.utils.instantiate(cfg.model)\n",
     "\n",
     "# Load pretrained weights from a checkpoint file\n",
-    "model = model.load_from_checkpoint(cfg.ckpt_path, net=model.net, criterion=None)\n",
-    "model.criterion = hydra.utils.instantiate(cfg.model).criterion\n",
+    "model = model.load_from_checkpoint(cfg.ckpt_path, net=model.net, criterion=model.criterion)\n",
     "model = model.eval().cuda()"
    ]
   },
@@ -370,4 +369,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}
diff --git a/notebooks/demo_kitti360.ipynb b/notebooks/demo_kitti360.ipynb
@@ -85,8 +85,7 @@
     "model = hydra.utils.instantiate(cfg.model)\n",
     "\n",
     "# Load pretrained weights from a checkpoint file\n",
-    "model = model.load_from_checkpoint(cfg.ckpt_path, net=model.net, criterion=None)\n",
-    "model.criterion = hydra.utils.instantiate(cfg.model).criterion\n",
+    "model = model.load_from_checkpoint(cfg.ckpt_path, net=model.net, criterion=model.criterion)\n",
     "model = model.eval().cuda()"
    ]
   },

diff --git a/notebooks/demo_s3dis.ipynb b/notebooks/demo_s3dis.ipynb
@@ -85,8 +85,7 @@
     "model = hydra.utils.instantiate(cfg.model)\n",
     "\n",
     "# Load pretrained weights from a checkpoint file\n",
-    "model = model.load_from_checkpoint(cfg.ckpt_path, net=model.net, criterion=None)\n",
-    "model.criterion = hydra.utils.instantiate(cfg.model).criterion\n",
+    "model = model.load_from_checkpoint(cfg.ckpt_path, net=model.net, criterion=model.criterion)\n",
     "model = model.eval().cuda()"
    ]
   },
@@ -370,4 +369,4 @@
  },
  "nbformat": 4,
  "nbformat_minor": 5
-}
+}