fix checkpoint downloading

NitCoh · Sep 1, 2020 · 1b62840 · 1b62840
1 parent 88d3896
commit 1b62840
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/download_pretrained.sh b/scripts/download_pretrained.sh
@@ -10,5 +10,6 @@ fi
 BLOB='https://convaisharables.blob.core.windows.net/uniter'
 
 for MODEL in uniter-base uniter-large; do
-    wget $BLOB/pretrained/$MODEL.pt -P $DOWNLOAD/pretrained/
+    # This will overwrite models
+    wget $BLOB/pretrained/$MODEL.pt -O $DOWNLOAD/pretrained/$MODEL.pt
 done