## Problem <!-- A clear and concise description of what the problem is. --> Currently there's a `ValueError` if `self.model.output_channels != 1`
Problem
Currently there's a
ValueErrorifself.model.output_channels != 1