From 08029852d9a23410a24363fc3af8cce00b78ab03 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 18 Sep 2020 22:14:51 -0700 Subject: [PATCH] Sotabench debugging --- sotabench.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/sotabench.py b/sotabench.py index 93c15d76..bbb75280 100644 --- a/sotabench.py +++ b/sotabench.py @@ -443,6 +443,12 @@ model_list = [ ] +# FIXME debug sotabench dataset issues +from pprint import pprint +from glob import glob +pprint([glob('./.data/**', recursive=True)]) +pprint([glob('./.data/vision/**', recursive=True)]) + for m in model_list: model_name = m['model'] # create model from name @@ -466,7 +472,7 @@ for m in model_list: input_transform=input_transform, batch_size=m['batch_size'], num_gpu=NUM_GPU, - data_root=os.environ.get('IMAGENET_DIR', './imagenet') + data_root=os.environ.get('IMAGENET_DIR', './.data/vision/imagenet') ) torch.cuda.empty_cache()