Fix: adding import nececeary for using attention layers as bottlenecks

walkwithfastai · Jun 20, 2024 · 2bb20d9 · 2bb20d9
1 parent ecdd5ed
commit 2bb20d9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nbs/02_vision.external.timm.ipynb b/nbs/02_vision.external.timm.ipynb
@@ -155,6 +155,7 @@
    "source": [
     "#export\n",
     "from timm import create_model\n",
+    "from fastai.layers import _get_norm\n",
     "from fastai.vision.learner import _update_first_layer"
    ]
   },