I run this notebook in Google colab and got an error when using simply the model name as the input to create_body function. Since children can be called from the function form, I added the parentheses in the notebook, so encoder = create_body(resnet34, cut=-2) became encoder = create_body(resnet34(), cut=-2) and the notebook runs fine.
Except it doesn't train up to 94% accuracy.
I run this notebook in Google colab and got an error when using simply the model name as the input to
create_body
function. Sincechildren
can be called from the function form, I added the parentheses in the notebook, soencoder = create_body(resnet34, cut=-2)
becameencoder = create_body(resnet34(), cut=-2)
and the notebook runs fine. Except it doesn't train up to 94% accuracy.