search

thuml / Self-Tuning

Code release for "Self-Tuning for Data-Efficient Deep Learning" (ICML 2021)
110 stars 14 forks source link

No module named 'efficientnet_pytorch' #7

Closed Fanyan-0411 closed 2 years ago

Fanyan-0411 commented 3 years ago

No module or files named 'efficientnet_pytorch' in the whole file holder. But 'from efficientnet_pytorch import EfficientNet' appears in the first line of efficientnet.py Is there any thing should be done before running main.py? Thanks for your reply!

wxm17 commented 2 years ago

You can easily install it by running pip install efficientnet_pytorch as shown in https://github.com/lukemelas/EfficientNet-PyTorch