github/tinygrad
1
1
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-04-29 03:00:14 -04:00
Code Issues Packages Projects Releases Wiki Activity

oops

Browse Source
This commit is contained in:
David Hou
2024-02-09 12:36:49 -08:00
parent c84671082d
commit cebcc4c703
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extra/datasets/imagenet.py
View File

@@ -16,7 +16,7 @@ def get_imagenet_categories():
def _get_train_files(): return glob.glob(str(BASEDIR / "train/*/*"))
@functools.lru_cache(None)
def get_train_files():
train_files = _get_train_files
train_files = _get_train_files()
# test train with less categories
if getenv("TEST_CATS", 1000) != 1000:
ci = get_imagenet_categories()