medicaldetectiontoolkit icon indicating copy to clipboard operation
medicaldetectiontoolkit copied to clipboard

Published 20 hours ago • verified publisher icon MIC-DKFZ

fix indexerror for retinanet in torch1.x

Open alibool opened this issue 2 years ago • 0 comments

keep_ix / fg_probs.shape[1] returns a float tensor which can not be used as index with torch 1.x.

alibool avatar Mar 14 '22 12:03 alibool