### Expected behavior I think the model should start training ### Actual behavior Error when trying to build the model with/without the pre-trained weights (following the building instruction) ### Steps to reproduce behavior  