search

JaryHuang / awesome_SSD_FPN_GIoU

This repository carries out some paper recurring work
106 stars 25 forks source link

copy_if failed to synchronize: cudaErrorAssert: device-side assert triggered #18

Open sankexin opened 4 years ago

sankexin commented 4 years ago

solve this problem: modify conf = labels[best_truth_idx] + 1 # Shape: [num_priors] to conf = labels[best_truth_idx] # Shape: [num_priors]