add comment line for running FM instead of ICFM in `conditional_mnist…

….ipynb`
atong01 · Jan 11, 2024 · ef9fb89 · ef9fb89
1 parent 3f9d813
commit ef9fb89
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/images/conditional_mnist.ipynb b/examples/images/conditional_mnist.ipynb
@@ -65,6 +65,7 @@
     "    dim=(1, 28, 28), num_channels=32, num_res_blocks=1, num_classes=10, class_cond=True\n",
     ").to(device)\n",
     "optimizer = torch.optim.Adam(model.parameters())\n",
+    "# FM = TargetConditionalFlowMatcher(sigma=sigma)\n",
     "FM = ConditionalFlowMatcher(sigma=sigma)\n",
     "node = NeuralODE(model, solver=\"dopri5\", sensitivity=\"adjoint\", atol=1e-4, rtol=1e-4)"
    ]