mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-06-24 01:17:28 +08:00
Use stable url for MNIST (#749)
* use stable url * remove deprecated flag
This commit is contained in:
parent
d1c35fa684
commit
2bf11c4633
@ -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
|
||||||
|
@ -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,
|
||||||
|
Loading…
Reference in New Issue
Block a user