2024年10月12日
用nni进行模型剪枝的示例
摘要： 第一步：模型训练 用下面的代码训练一个简单的模型：（数据集参考：利用pytorch的datasets在本地读取MNIST数据集进行分类） import torch import torch.nn as nn import torch.optim as optim from torchvision i 阅读全文
posted @ 2024-10-12 14:11 Picassooo 阅读(214) 评论(0) 推荐(0)
利用pytorch的datasets在本地读取MNIST数据集进行分类
摘要： MNIST数据集下载地址：tensorflow-tutorial-samples/mnist/data_set at master · geektutu/tensorflow-tutorial-samples · GitHub 数据集存放和dataset的参数设置： 完整的MNIST分类代码： im 阅读全文
posted @ 2024-10-12 10:53 Picassooo 阅读(616) 评论(0) 推荐(0)