Spaces:

JuyeopDang
/

KoFace-AI

Sleeping

KoFace-AI / auto_encoder /components /normalize.py

Upload 35 files

5ab5cab verified over 1 year ago

172 Bytes

import torch

def Normalize(in_channels : int, num_groups : int = 32):
    return torch.nn.GroupNorm(num_groups=num_groups, num_channels=in_channels, eps=1e-6, affine=True)

	import torch

	def Normalize(in_channels : int, num_groups : int = 32):
	return torch.nn.GroupNorm(num_groups=num_groups, num_channels=in_channels, eps=1e-6, affine=True)