add instructions in conditional_mnist.ipynb for running FM

atong01 · Jan 11, 2024 · 06702c8 · 06702c8
1 parent 3f9d813
commit 06702c8
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)"
    ]
   },