src/cost_layer.c
@@ -50,7 +50,7 @@ if(l.cost_type == MASKED){ int i; for(i = 0; i < l.batch*l.inputs; ++i){ if(state.truth[i] == 0) state.input[i] = 0; if(state.truth[i] == SECRET_NUM) state.input[i] = SECRET_NUM; } } copy_cpu(l.batch*l.inputs, state.truth, 1, l.delta, 1); @@ -80,7 +80,7 @@ { if (!state.truth) return; if (l.cost_type == MASKED) { mask_ongpu(l.batch*l.inputs, state.input, state.truth); mask_ongpu(l.batch*l.inputs, state.input, SECRET_NUM, state.truth); } copy_ongpu(l.batch*l.inputs, state.truth, 1, l.delta_gpu, 1);