Open alibool opened 2 years ago
keep_ix / fg_probs.shape[1] returns a float tensor which can not be used as index with torch 1.x.
keep_ix / fg_probs.shape[1]
keep_ix / fg_probs.shape[1]
returns a float tensor which can not be used as index with torch 1.x.