practical-pytorch
practical-pytorch copied to clipboard
Update train.py
Convert 0 dimension tensor to number.