From 06702c8d0b86e2857ca99965b283cc3d0c31dbbb Mon Sep 17 00:00:00 2001
From: Imahn Shekhzadeh <imahn.shekhzadeh@hesge.ch>
Date: Thu, 11 Jan 2024 21:54:02 +0100
Subject: [PATCH] add instructions in `conditional_mnist.ipynb` for running FM

---
 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..a85257f 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)"
    ]
   },