practical-pytorch icon indicating copy to clipboard operation
practical-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon spro

Update train.py

Open mtfelix opened this issue 6 years ago • 0 comments

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

mtfelix avatar Sep 06 '19 03:09 mtfelix