Notebook on Efficientnet runs into error on Colab
ne1s0n opened this issue · comments
With reference to file 05_EfficientNet_and_Custom_Weights.ipynb, once the learner is created, albeit not frozen, a call to
learn.summary()
generates
RuntimeError: running_mean should contain 3072 elements not 6144
instead of showing the proper summary. If ignored the later call to learn.lr_find()
stops with the same error. I was running the code on Google Colab.
Can you try running the notebook from here?
https://walkwithfastai.com/EfficientNet_and_Custom_Weights
The notebooks are updated at https://github.com/walkwithfastai/walkwithfastai.github.io
Same result - you can see a screenshot here: https://ibb.co/Lg9PzWB
I've actually found the error on walkwithfastai and then came here because you linked this repo in the fastai forum :)