Fixed as per the solution in https://forums.fast.ai/t/chapter-15-fastai-book-issue-siamese-training-running-mean/84534