- 博客(3)
- 收藏
- 关注
原创 微调预训练的vgg16 网络
链接:https://pan.baidu.com/s/1ObAOosSVPJ6iB0fQUojJHA?注:新版本vgg16中没有pretrained参数,可根据weights参数选择网络。注:train_dir和val_dir根据数据集文件路径位置填写。步骤3 可视化一个batch的数据(可注释掉)--来自百度网盘超级会员V3的分享。步骤6 微调网络的训练和预测。步骤4 导入VGG16网络。步骤1 预处理数据集。步骤5 搭建神经网络。
2024-07-21 00:47:47 223
原创 FashionMNIST数据集及神经网络搭建
上一篇文章犯了个错误,图像不归一化的训练效果比归一化要好,也就是说无需除255.0。模型预测率达到80%以上。
2024-07-18 11:41:02 294
原创 pytorch网络搭建和MNIST数据集训练代码
可视化在tensorboard里看,终端输入 tensorboard --logdir="文件名",然后打开它给的网址。1.transforms.Totensor并未归一化训练集,因此额外除了255.0,以此归一化。2.tensorboardX安装时先安装tensorboard,再安装tensorboardX。3.标签独热编码后出现valueerror,也没解决。4.pytorch bug太多,不太好用。
2024-07-16 16:41:58 292
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人