Ejemplo n.º 1
0
 def add(module,
         affine=True,
         momentum=0.9,
         epsilon=1e-5,
         fuse=None,
         use_out_hook=True):
     Norm.add(module,
              RunningNorm(affine, momentum, epsilon, fuse, use_out_hook),
              use_out_hook=use_out_hook)
 def add(module,
         affine=True,
         momentum=0.9,
         epsilon=1e-5,
         track_statistics=True,
         fuse=None,
         use_out_hook=True):
     Norm.add(module,
              BatchNorm(affine, momentum, epsilon, track_statistics, fuse,
                        use_out_hook),
              use_out_hook=use_out_hook)