Update Interactive_Back_Translation.ipynb

vietai · Mar 11, 2021 · 728793b · 728793b
1 parent 96716f6
commit 728793b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/colab/Interactive_Back_Translation.ipynb b/colab/Interactive_Back_Translation.ipynb
@@ -10,8 +10,8 @@
  "include_colab_link": true
  },
  "kernelspec": {
- "name": "python2",
- "display_name": "Python 2"
+ "name": "python3",
+ "display_name": "Python 3"
  },
  "accelerator": "GPU"
  },
@@ -134,7 +134,9 @@
  "colab": {}
  },
  "source": [
+ "%tensorflow_version 1.x\n",
  "!pip install -q -U tensor2tensor\n",
+ "!pip install tensorflow-datasets==3.2.1\n",
  "\n",
  "import os\n",
  "from tensor2tensor.bin import t2t_decoder\n",
@@ -305,4 +307,4 @@
  ]
  }
  ]
-}
+}