UPDATE: While I’ve not exactly found the solution, I think I have found the cause of the issue. I was in effect trying to load pre-trained models and apply them. These models must’ve been trained with an older version of pytorch, where BatchNorm2d did not have (and hence did not save) the ‘track_running_stats’ attribute.
The confusing thing is that using an older version of pytorch failed due to a reliance on the ‘_rebuild_tensor_v2’ attribute of torch._utils, which seemingly references a pytorch version later than that used to train the example models.
My only explanation is that the models available were saved with an older version of pytorch, development was then made with a newer version of pytorch, and backwards compatibility with available pre-trained models was not checked.
The fix for me is hopefully just to re-train all models from scratch using a new version of pytorch, so that my own saved models will have BatchNorm2d objects with the track_runnning_stats attributes (c.f. https://github.com/xingyizhou/pytorch-pose-hg-3d/issues/36 8).
Thanks!