def __init__(self,
output_num_channels=2,
input_num_channels=3,
inner_nc=128):
super(BottleneckFromNet, self).__init__()
self.pix2pixSampler = NoSkipPix2PixModel(input_num_channels,
output_num_channels,
inner_nc=inner_nc)
def __init__(self,
output_num_channels=2,
input_num_channels=3,
inner_nc=512):
super(UnwrappedFaceWeightedAverage, self).__init__()
self.pix2pixUnwrapped = Pix2PixModel(3)
self.pix2pixSampler = NoSkipPix2PixModel(input_num_channels,
output_num_channels,
inner_nc=inner_nc)