91e003f969109dce7f2274e6f7145a4e9dbc4ef9,chainercv/links/model/light_head_rcnn/light_head_rcnn_train_chain.py,,_ohem_loss,#Any#Any#Any#Any#Any#Any#,188

Before Change

    n_ohem_sample = min(n_ohem_sample, n_sample)
    // sort in CPU because of GPU memory
    roi_cls_loc_loss = cuda.to_cpu(roi_loc_loss.array + roi_cls_loss.array)
    indices = roi_cls_loc_loss.argsort(axis=0)[::-1][:n_ohem_sample]
    if cuda.get_array_module(roi_loc_loss.array) != np:
        indices = cuda.to_gpu(indices)
    roi_loc_loss = F.sum(roi_loc_loss[indices]) / len(indices)
    roi_cls_loss = F.sum(roi_cls_loss[indices]) / len(indices)

After Change

    roi_cls_loc_loss = cuda.to_cpu(roi_loc_loss.array + roi_cls_loss.array)
    indices = roi_cls_loc_loss.argsort(axis=0)[::-1]
    // filter nan
    indices = np.array(
        [i for i in indices if not np.isnan(roi_cls_loc_loss[i])],
        dtype=np.int32)
    indices = indices[:n_ohem_sample]
    if cuda.get_array_module(roi_loc_loss.array) != np:
        indices = cuda.to_gpu(indices)
    if len(indices) > 0:

Instances