Skip to content

Commit

Permalink
Update gdrive links
Browse files Browse the repository at this point in the history
  • Loading branch information
dorarad authored May 27, 2022
1 parent 13d9bcf commit 8362139
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions pytorch_version/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,17 +7,18 @@
import dnnlib
from torch_utils import misc


pretrained_networks = {
"gdrive:clevr-snapshot.pkl": "https://drive.google.com/uc?id=1zBh-U2kyVgN3C_P_7GqsMEHvBdz2lobu",
"gdrive:cityscapes-snapshot.pkl": "https://drive.google.com/uc?id=1XPGYzUP_1ETFtz5bUhpUFPha1IBNTuZh",
"gdrive:ffhq-snapshot.pkl": "https://drive.google.com/uc?id=1tgs-hHaziWrh0piuX3sEd8PwE9gFwlNh",
"gdrive:bedrooms-snapshot.pkl": "https://drive.google.com/uc?id=19TykSlMgXIjyIiDLBakmNVEQmD5dTd0s"
"gdrive:clevr-snapshot.pkl": "https://drive.google.com/uc?id=1sYtuNEi0HGBH1F8GW2JlF4mCRGGrtZa9",
"gdrive:cityscapes-snapshot.pkl": "https://drive.google.com/uc?id=1iL8S26IYCNAQmUS-6GYCVheKFtCE4ktQ",
"gdrive:ffhq-snapshot.pkl": "https://drive.google.com/uc?id=1-b0vwevUQs6LI_EybdO8XJ5uYSx63vEa",
"gdrive:bedrooms-snapshot.pkl": "https://drive.google.com/uc?id=1lrB4t3hOOpn8YiwHEduIZYzndHrsKX8z"
}

eval_pretrained_networks = pretrained_networks.copy()
eval_pretrained_networks.update({
# "gdrive:cityscapes-snapshot-2048.pkl": "https://drive.google.com/uc?id=1Zw1cFxxN6-iC_M4x6Zbf9lwH9wKryW3p",
"gdrive:ffhq-snapshot-1024.pkl": "https://drive.google.com/uc?id=10V4yK_rQWb4F6Q4vwqkO5XNKX721k3zl"
"gdrive:ffhq-snapshot-1024.pkl": "https://drive.google.com/uc?id=1OcsibUthk2y8y0slf2lQnrYBAtmdJYCQ"
})

def get_path_or_url(path_or_gdrive_path, eval = False):
Expand Down

0 comments on commit 8362139

Please sign in to comment.