摘要： Top-1 error 的意思是：假如模型预测某张动物图片（一只猫）的类别，且模型只输出1个预测结果，那么这一个结果正好能猜出来这个动物是只猫的概率就是Top-1正确率。 猜出来的结果不是猫的概率则成为Top-1错误率。简单来说就是模型猜错的概率。 Top-5 error 的意思是：假如模型预测某张 阅读全文

摘要： 1、保存文件的后缀 有.pt，.pth，.pkl，在格式上没有任何区别，都能用，随便哪种都行。 2、保存和加载整个模型 torch.save(net, 'net_model.pkl') 加载模型 net_loaded = torch.load('net_model.pkl') 3、只保存模型参数 t 阅读全文