changing removed loss.data[0] to loss.item()#1

Open

Samarth26 wants to merge 1 commit intocmubig:masterfrom Samarth26:master

+2-2

Commits on Apr 8, 2024

changing deprecated loss.data[0] to loss.item()
Samarth26
committed