muellerzr / Practical-Deep-Learning-for-Coders-2.0

Notebooks for the "A walk with fastai2" Study Group and Lecture Series

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

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 :)