This forum post provides a solution which is to change the following line in the notebook (seems like a version issue with timm):
path = setup_comp(comp, install='fastai "timm>=0.6.2.dev0"')
to
path = setup_comp(comp, install='fastai "timm==0.6.2.dev0"')
I just tested it out and was able to load and train the model with that change:
