pretrained-models.pytorch icon indicating copy to clipboard operation
pretrained-models.pytorch copied to clipboard
verified publisher icon Cadene

fix 0-dim tensor error

Open fangrenfu opened this issue 5 years ago • 0 comments

Use tensor.item() to convert a 0-dim tensor to a Python number

fangrenfu avatar Mar 12 '20 16:03 fangrenfu