Skip to content

GitLab

Unverified Commit ad3bbef9 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

fashion_mnist (#3337)

parent d5bc2b24
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
test/test_datasets_download.py
View file @ ad3bbef9
...@@ -195,6 +195,10 @@ def mnist(): ...@@ -195,6 +195,10 @@ def mnist():
return collect_download_configs(lambda: datasets.MNIST(".", download=True), name="MNIST") return collect_download_configs(lambda: datasets.MNIST(".", download=True), name="MNIST")
def fashion_mnist():
return collect_download_configs(lambda: datasets.FashionMNIST(".", download=True), name="FashionMNIST")
def make_parametrize_kwargs(download_configs): def make_parametrize_kwargs(download_configs):
argvalues = [] argvalues = []
ids = [] ids = []
...@@ -216,6 +220,7 @@ def make_parametrize_kwargs(download_configs): ...@@ -216,6 +220,7 @@ def make_parametrize_kwargs(download_configs):
# The VOC download server is unstable. See https://github.com/pytorch/vision/issues/2953 for details. # The VOC download server is unstable. See https://github.com/pytorch/vision/issues/2953 for details.
# voc(), # voc(),
mnist(), mnist(),
fashion_mnist(),
) )
) )
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment