Added paired_desc option in the dataset visualization notebook

cvg · May 5, 2023 · bbce15c · bbce15c
1 parent 3d7bcd6
commit bbce15c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/notebooks/visualize_exported_dataset.ipynb b/notebooks/visualize_exported_dataset.ipynb
@@ -48,6 +48,7 @@
     "# Initialize the wireframe dataset\n",
     "with open(\"../sold2/config/wireframe_dataset.yaml\", \"r\") as f:\n",
     "    config = yaml.safe_load(f)\n",
+    "config['return_type'] = 'paired_desc'\n",
     "\n",
     "wireframe_dataset = WireframeDataset(mode=\"test\", config=config)"
    ]
@@ -390,7 +391,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.7.5"
+   "version": "3.6.9"
   },
   "mimetype": "text/x-python",
   "name": "python",