diff --git a/15_arch_details.ipynb b/15_arch_details.ipynb index 2189794..ee9d847 100644 --- a/15_arch_details.ipynb +++ b/15_arch_details.ipynb @@ -343,7 +343,7 @@ "metadata": {}, "outputs": [], "source": [ - "encoder = create_body(resnet34, cut=-2)" + "encoder = create_body(resnet34(), cut=-2)" ] }, { diff --git a/clean/15_arch_details.ipynb b/clean/15_arch_details.ipynb index 4a9af48..906210f 100644 --- a/clean/15_arch_details.ipynb +++ b/clean/15_arch_details.ipynb @@ -147,7 +147,7 @@ "metadata": {}, "outputs": [], "source": [ - "encoder = create_body(resnet34, cut=-2)" + "encoder = create_body(resnet34(), cut=-2)" ] }, {