Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Improve download scripts #133

Merged
merged 3 commits into from
Jun 29, 2023

Conversation

kevinunger
Copy link
Contributor

The download scripts in scripts/ do not do error handling, are a bit redundant and don't show the download progress. Also there needs to be scripts for windows and linux.
This PR incudes a download script that:

  • works cross-platform
  • prints the progress and connection info
  • skips already downloaded files.
  • all checkpoints names and urls are defined in a .json

Output while downloading:

C:\Users\kevin\Gits\forks\DragGAN>python scripts/download_model.py
checkpoints\stylegan2_lions_512_pytorch.pkl already exists but size does not match. Redownloading.
100%|█████████████████████████████| 364M/364M [01:04<00:00, 5.68MiB/s]
Downloading stylegan2_dogs_1024_pytorch.pkl from https://storage.googleapis.com/self-distilled-stylegan/dogs_1024_pytorch.pkl
  4%|| 14.6M/382M [00:02<01:03, 5.79MiB/s]

Unverified

This commit is not signed, but one or more authors requires that any commit attributed to them is signed.

Unverified

This commit is not signed, but one or more authors requires that any commit attributed to them is signed.
@XingangPan XingangPan merged commit 1a2d37d into XingangPan:main Jun 29, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

2 participants