pred=logits.argmax(dim=1) num_correct+=torch.eq(pred, y).sum().float().item() logger.info("Train Epoch: {}\t Loss: {:.6f}\t Acc: {:.6f}".format(epoch,train_loss/len(train_loader),num_correct/len(train_loader.dataset))) returnnum_correct/len(train_loader.dataset), train_los...
www.dbjr.com.cn/article/1697...htm 2025-6-8