mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Sotabench debugging

Browse Source
This commit is contained in:
Ross Wightman 2020-09-18 22:14:51 -07:00
parent c40384f5bd
commit 08029852d9
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
sotabench.py
View File

@ -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()