685bf12edbaf20d969dc28549e634c9fa4993f5f,dl/callbacks.py,IOUCallback,on_batch_end,#IOUCallback#Any#,98

Before Change

        valid_msks = state.input[self.input_key]

        iou = get_iou_vector(valid_msks.detach().cpu().numpy(),
                       msk_vpreds.sigmoid().detach().cpu().numpy())
        key = "iou"
        state.batch_metrics[key] = iou

After Change

        valid_msks = state.input[self.input_key]

        // msk_vpreds = msk_vpreds.sigmoid()
        msk_vpreds = msk_vpreds.detach().cpu().numpy()
        valid_msks = valid_msks.detach().cpu().numpy()

        iou = get_iou_vector(valid_msks, msk_vpreds)
        key = "iou"
        state.batch_metrics[key] = iou

Instances