@ptrblck Hi I’m recreating an implementation(LITEFLOW-NET)
so what I’m doing is using the MSELoss since the out put is a tensor of shape(2,436,1024) with real values.
I’can’t think of a better loss function for this case