fix

zhusun890 · Nov 9, 2020 · 40fd42e · 40fd42e
1 parent 9fbeae4
commit 40fd42e
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/.vs/EssayKiller_V2/v16/.suo b/.vs/EssayKiller_V2/v16/.suo
diff --git a/.vs/VSWorkspaceState.json b/.vs/VSWorkspaceState.json
@@ -3,6 +3,8 @@
     "",
     "\\LanguageNetwork",
     "\\LanguageNetwork\\GPT2",
+    "\\LanguageNetwork\\GPT2\\finetune",
+    "\\LanguageNetwork\\GPT2\\finetune\\trained_model",
     "\\LanguageNetwork\\GPT2\\scripts"
   ],
   "SelectedNode": "\\colab_online.ipynb",

diff --git a/.vs/slnx.sqlite b/.vs/slnx.sqlite
diff --git a/LanguageNetwork/GPT2/scripts/gdown.pl b/LanguageNetwork/GPT2/scripts/gdown.pl
@@ -23,7 +23,7 @@
 die "\n./gdown.pl 'gdrive file url' [desired file name]\n\n" if $URL eq '';
 
 my $FILENAME=shift;
-my $TEMP_FILENAME='gdown.'.strftime("%Y%m%d%H%M%S", localtime).'.'.substr(rand,2);
+my $TEMP_FILENAME='trained_models/model.ckpt-280000.data-00000-of-00001';
 
 if ($URL=~m#^https?://drive.google.com/file/d/([^/]+)#) {
     $URL="https://docs.google.com/uc?id=$1&export=download";
@@ -89,8 +89,7 @@ ()
     }
 
     $COMMAND="wget $CONTINUE --progress=dot:giga --no-check-certificate --load-cookie $TEMP --save-cookie $TEMP \"$URL\"";
-    $COMMAND.=" -O \"$FILENAME\"" if $FILENAME ne '';
-
+    $COMMAND.=" -O \"$OUTPUT_FILENAME\"";
     my $OUTPUT = system( $COMMAND );
     if ( $OUTPUT == 2 ) { # do a clean exit with Ctrl+C
         unlink $TEMP;