Chapter 15 fastai book issue (Siamese training running mean)

Hello,

When executing the chapter 15 of the fastai book, I get an error I can’t quite figure out.
With no change whatsoever to the code of the colab workbout I get the following error:

RuntimeError: running_mean should contain 2048 elements not 4096

After executing the cell ====> learn.fit_one_cycle(4, 3e-3)

In case you need the link to the colab notebook for chapter 15.
Chapter 15, Arch Details

Set the inputs to create_head in half (so rather than 512x4, do 512x2)

2 Likes

That does the trick, thank you very much!

I do understand why this solves the error but I do not fully understand why we have to change this? In the fastbook the *4 is explained because: 2 images, 2 poolingLayers --> 4. Was there any change in the fastai framework?

Yes (hence the solution). Now create_head will setup the proper input filters based on if we’re doing two pooling layers or not (so it will do that * 2 internally)

1 Like