add comment lines for running FM instead of ICFM in `conditional_mnis…

…t.ipynb` (#97)
atong01 · Jan 15, 2024 · 7898ff1 · 7898ff1
1 parent 3f9d813
commit 7898ff1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/examples/images/conditional_mnist.ipynb b/examples/images/conditional_mnist.ipynb
@@ -66,6 +66,8 @@
     ").to(device)\n",
     "optimizer = torch.optim.Adam(model.parameters())\n",
     "FM = ConditionalFlowMatcher(sigma=sigma)\n",
+    "# Users can try target FM by changing the above line by\n",
+    "# FM = TargetConditionalFlowMatcher(sigma=sigma)\n",
     "node = NeuralODE(model, solver=\"dopri5\", sensitivity=\"adjoint\", atol=1e-4, rtol=1e-4)"
    ]
   },