Summary: 4 instances, 4 unique

Text	Count
# FIXME this may need to be more generic / flexible for some nets	1
FIXME this does not currently work with Torchscript, see FeatureHooks class	1
# TODO: change the name to `norm`	1
# TODO: merge update_prediction with update_stats.	1