From 7898ff185c0bb3d627e7e01a476c21f8f9a86f83 Mon Sep 17 00:00:00 2001
From: Imahn Shekhzadeh <120030096+ImahnShekhzadeh@users.noreply.github.com>
Date: Mon, 15 Jan 2024 23:57:09 +0100
Subject: [PATCH] add comment lines for running FM instead of ICFM in
 `conditional_mnist.ipynb` (#97)

---
 examples/images/conditional_mnist.ipynb | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/examples/images/conditional_mnist.ipynb b/examples/images/conditional_mnist.ipynb
index c5b19b8..f2e4cf1 100644
--- 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)"
    ]
   },