Use stable url for MNIST (#749)

* use stable url

* remove deprecated flag
This commit is contained in:
Awni Hannun 2024-05-03 17:13:05 -07:00 committed by GitHub
parent d1c35fa684
commit 2bf11c4633
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 3 additions and 2 deletions

View File

@ -80,7 +80,6 @@ def get_model_path(path_or_hf_repo: str, revision: Optional[str] = None) -> Path
"*.tiktoken", "*.tiktoken",
"*.txt", "*.txt",
], ],
resume_download=True,
) )
) )
return model_path return model_path

View File

@ -9,7 +9,9 @@ import numpy as np
def mnist( def mnist(
save_dir="/tmp", base_url="http://yann.lecun.com/exdb/mnist/", filename="mnist.pkl" save_dir="/tmp",
base_url="https://raw.githubusercontent.com/fgnt/mnist/master/",
filename="mnist.pkl",
): ):
""" """
Load the MNIST dataset in 4 tensors: train images, train labels, Load the MNIST dataset in 4 tensors: train images, train labels,