Fix download pretrain faceset

CelebA is now downloaded directly from the Github repository. P.S. gdrive suxx
tania010 · Feb 24, 2020 · f3adc66 · f3adc66
1 parent cf6403e
commit f3adc66
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/DFL_Colab.ipynb b/DFL_Colab.ipynb
@@ -120,8 +120,8 @@
         "!pip install --upgrade scikit-image\n",
         "!apt-get install cuda-10-0\n",
         "\n",
-        "if not Path(\"/content/workspace\").exists():\n",
-        "  !wget -q --no-check-certificate -r 'https://docs.google.com/uc?export=download&id=1hTH2h6l_4kKrczA8EkN6GyuXx4lzmCnK' -O pretrain_CelebA.zip\n",
+        "if not Path(\"/content/pretrain\").exists():\n",
+        "  !wget -q --no-check-certificate -r 'https://github.com/chervonij/DFL-Colab/releases/download/pretrain-CelebA/pretrain_CelebA.zip' -O pretrain_CelebA.zip\n",
         "  !mkdir /content/pretrain\n",
         "  !unzip -q /content/pretrain_CelebA.zip -d /content/pretrain/\n",
         "  !rm /content/pretrain_CelebA.zip\n",
@@ -592,7 +592,7 @@
         "  print(\"Session expires in less than an hour.\")\n",
         "  backup_cmd = \"\"\n",
         "    \n",
-        "cmd = \"DeepFaceLab/main.py train --training-data-src-dir workspace/data_src/aligned --training-data-dst-dir workspace/data_dst/aligned --pretraining-data-dir pretrain/aligned --model-dir workspace/model --model \"+Model\n",
+        "cmd = \"DeepFaceLab/main.py train --training-data-src-dir workspace/data_src/aligned --training-data-dst-dir workspace/data_dst/aligned --pretraining-data-dir pretrain --model-dir workspace/model --model \"+Model\n",
         "  \n",
         "if (backup_cmd != \"\"):\n",
         "  train_cmd = (cmd+backup_cmd)\n",

diff --git a/DFL_Colab_1-0.ipynb b/DFL_Colab_1-0.ipynb
@@ -120,8 +120,8 @@
         "!pip install --upgrade scikit-image\n",
         "!apt-get install cuda-10-0\n",
         "\n",
-        "if not Path(\"/content/workspace\").exists():\n",
-        "  !wget -q --no-check-certificate -r 'https://docs.google.com/uc?export=download&id=1hTH2h6l_4kKrczA8EkN6GyuXx4lzmCnK' -O pretrain_CelebA.zip\n",
+        "if not Path(\"/content/pretrain\").exists():\n",
+        "  !wget -q --no-check-certificate -r 'https://github.com/chervonij/DFL-Colab/releases/download/pretrain-CelebA/pretrain_CelebA.zip' -O pretrain_CelebA.zip\n",
         "  !mkdir /content/pretrain\n",
         "  !unzip -q /content/pretrain_CelebA.zip -d /content/pretrain/\n",
         "  !rm /content/pretrain_CelebA.zip\n",
@@ -607,7 +607,7 @@
         "  print(\"Session expires in less than an hour.\")\n",
         "  backup_cmd = \"\"\n",
         "    \n",
-        "cmd = \"DeepFaceLab/main.py train --training-data-src-dir workspace/data_src/aligned --training-data-dst-dir workspace/data_dst/aligned --pretraining-data-dir pretrain/aligned --model-dir workspace/model --model \"+Model\n",
+        "cmd = \"DeepFaceLab/main.py train --training-data-src-dir workspace/data_src/aligned --training-data-dst-dir workspace/data_dst/aligned --pretraining-data-dir pretrain --model-dir workspace/model --model \"+Model\n",
         "  \n",
         "if (backup_cmd != \"\"):\n",
         "  train_cmd = (cmd+backup_cmd)\n",