pipert.contrib.detection_demo.torch_utils

Module Contents

Functions

init_seeds(seed=0)
select_device(device=None, force_cpu=False, apex=False)
fuse_conv_and_bn(conv, bn)

pipert.contrib.detection_demo.torch_utils.init_seeds(seed=0)

pipert.contrib.detection_demo.torch_utils.select_device(device=None, force_cpu=False, apex=False)

pipert.contrib.detection_demo.torch_utils.fuse_conv_and_bn(conv, bn)