from model.registry import MODEL, NORMALIZATION import model.GAN import model.base.normalization