分类：模型准确率

2021
07-01

Pytorch中求模型准确率的两种方法小结

方法一：直接在epoch过程中求取准确率简介：此段代码是LeNet5中截取的。deftrain_model(model,train_loader):optimizer=torch.optim.Adam(model.parameters())loss_func=nn.CrossEntropyLoss()EPOCHS=5forepochinrange(EPOCHS):correct=0forbatch_idx,(X_batch,y_batch)inenumerate(train_loader):optimizer.zero_grad()#这里是只取训练数据的意思吗，X_b... 继续阅读 >