hi everybody,
i’m trying to learn parameters for forward and inverse gamma correction:
input->gamma_correction->network->another_gamma_correction->loss
where: gamma_correction(x,gamma_parameter)=x**gamma_parameter
what i do is simply define a nn.Parameter() for the initial correction and another nn.Parameter() for the final correction.
when i only try to learn the initial gamma correction there’s no problem but when i try to learn both initial and final gamma corrections i get nan values for the parameters after the first backpropagation.
any thought?
thanks