So i was doing something wrong with cuda on the server side.
To the weight parameter. It is good for me, but if i am understand it right way, it just do not count to the loss the output from the model that represents the ‘2’ class.
Do exists something, or is there some way, that loss would not be counted over this places where gt is ‘2’?
Now if i am understand what this parameter do, it is something like (class0OutputFromModel difference of Gt(where value is 0)) + (class1OutputFromModel difference of Gt(where value is 1)).
But at places where in GT is class ‘2’ i do not know if there should be class 1 or 2.
Or am i wrong and it is doing exactly what i want to achieve?