from fvcore.nn.precise_bn import get_bn_modules

[Trovias]

you need to add:
from fvcore.nn.precise_bn import get_bn_modules
in train.py and train_voc.py

[vidit09]

thanks