Update 03_transpiling_a_tf_model.ipynb

vaatsalya123 · Feb 21, 2024 · 3648ae4 · 3648ae4
1 parent d5dd40c
commit 3648ae4
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")"
       ]
     },
     {