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)