dgs.utils.torchtools.set_bn_to_eval¶

dgs.utils.torchtools.set_bn_to_eval(module: TorchMod | BaseMod) → None[source]¶

Sets BatchNorm layers to eval mode.