pytorch This is a pytorch model running over the channel of images and mapping them to a ground truth color conversion.