Peter Ross 27845fd3a3 train_efficientnet: only import datasets.imagenet when IMAGENET is set (#899) ...

make it work out of the box for new users.

the default configuration of train_efficientnet is to use the smaller cifar
dataset. import datasets.imagenet tries to open imagenet_class_index.json
and will fail, unless user has already downloaded it.

2023-06-01 19:19:52 -07:00

2.8 KiB

Raw Blame History

View Raw