Mirrors_Framework/deepmind-research
1
0
Fork 0
mirror of https://github.com/google-deepmind/deepmind-research.git synced 2026-05-20 03:11:31 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update PolyGen installation to do single pip install & clone into /tmp.

Browse Source 
This fixes issues with dep versioning resolution between each package.

PiperOrigin-RevId: 391343870
This commit is contained in:
Tom Ward
2021-08-17 19:57:46 +01:00
committed by Diego de Las Casas
parent 17b5dbaf88
commit 3d9f601c69
2 changed files with 12 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
polygen/sample-pretrained.ipynb
+6 -5
View File
@@ -42,9 +42,7 @@
},
"outputs": [],
"source": [
"!pip install tensorflow==1.15\n",
"!pip install dm-sonnet==1.36\n",
"!pip install tensor2tensor==1.14\n",
"!pip install tensorflow==1.15 dm-sonnet==1.36 tensor2tensor==1.14\n",
"\n",
"import time\n",
"import numpy as np\n",
@@ -52,8 +50,11 @@
"tf.logging.set_verbosity(tf.logging.ERROR) # Hide TF deprecation messages\n",
"import matplotlib.pyplot as plt\n",
"\n",
"!git clone https://github.com/deepmind/deepmind-research.git deepmind_research\n",
"%cd deepmind_research/polygen\n",
"%cd /tmp\n",
"%rm -rf /tmp/deepmind_research\n",
"!git clone https://github.com/deepmind/deepmind-research.git \\\n",
" /tmp/deepmind_research\n",
"%cd /tmp/deepmind_research/polygen\n",
"import modules\n",
"import data_utils"
]
polygen/training.ipynb
+6 -5
View File
@@ -42,9 +42,7 @@
},
"outputs": [],
"source": [
"!pip install tensorflow==1.15\n",
"!pip install dm-sonnet==1.36\n",
"!pip install tensor2tensor==1.14\n",
"!pip install tensorflow==1.15 dm-sonnet==1.36 tensor2tensor==1.14\n",
"\n",
"import os\n",
"import numpy as np\n",
@@ -52,8 +50,11 @@
"tf.logging.set_verbosity(tf.logging.ERROR) # Hide TF deprecation messages\n",
"import matplotlib.pyplot as plt\n",
"\n",
"!git clone https://github.com/deepmind/deepmind-research.git deepmind_research\n",
"%cd deepmind_research/polygen\n",
"%cd /tmp\n",
"%rm -rf /tmp/deepmind_research\n",
"!git clone https://github.com/deepmind/deepmind-research.git \\\n",
" /tmp/deepmind_research\n",
"%cd /tmp/deepmind_research/polygen\n",
"import modules\n",
"import data_utils"
]