fix

cleardusk · Aug 31, 2020 · 6077711 · 6077711
1 parent 1dd78e0
commit 6077711
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demo.ipynb b/demo.ipynb
@@ -42,7 +42,7 @@
    "outputs": [],
    "source": [
     "# load config\n",
-    "cfg = yaml.load(open('configs/mb1_120x120.yml'), Loader=yaml.FullLoader)\n",
+    "cfg = yaml.load(open('configs/mb1_120x120.yml'), Loader=yaml.SafeLoader)\n",
     "tddfa = TDDFA(gpu_mode=False, **cfg)\n",
     "\n",
     "# Initialize FaceBoxes\n",