Fix demo bug. (#75)

* Update 03_transpiling_a_tf_model.ipynb * Update 03_transpiling_a_tf_model.ipynb * Update 03_transpiling_a_tf_model.ipynb * Update 03_transpiling_a_tf_model.ipynb
ivy-llc · Feb 21, 2024 · 104f88a · 104f88a
1 parent d5dd40c
commit 104f88a
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/guides/03_transpiling_a_tf_model.ipynb b/guides/03_transpiling_a_tf_model.ipynb
@@ -119,8 +119,9 @@
       "outputs": [],
       "source": [
         "# Transpile it into a torch.nn.Module with the corresponding parameters\n",
+        "ivy.set_soft_device_mode(True)\n",
         "noise = tf.random.normal(shape=(1, 224, 224, 3))\n",
-        "torch_eff_encoder = ivy.transpile(eff_encoder, to=\"torch\", args=(noise,))"
+        "torch_eff_encoder = ivy.transpile(eff_encoder, source=\"tensorflow\", to=\"torch\", args=(noise,), modes_to_trace=\"eval\")"
       ]
     },
     {