Teja
Results
1
comments of
Teja
> avg_loss += loss.data[0] Replace avg_loss += loss.data[0] with avg_loss += loss.item()