def to_rgb(in_channels): #TODO: Make this respect 1 or 3 channels outchannels = 1 #return _equalized_conv2d(in_channels, 3, (1, 1), bias=True) return _equalized_conv2d(in_channels, outchannels, (1, 1), bias=True)
def to_rgb(in_channels): return _equalized_conv2d(in_channels, 3, (1, 1), bias=True)
def from_rgb(out_channels): #TODO: Make this respect 1 or 3 channels inchannels = 1 return _equalized_conv2d(inchannels, out_channels, (1, 1), bias=True)
def from_rgb(out_channels): return _equalized_conv2d(3, out_channels, (1, 1), bias=True)