Sotabench debugging

pull/244/head
Ross Wightman 4 years ago
parent c40384f5bd
commit 08029852d9

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

Loading…
Cancel
Save