随笔分类 -  pytorch

神经网络框架值pytorch，速度很快。
pytorch 实现 AlexNet 网络模型训练自定义图片分类
摘要：1、AlexNet网络模型，pytorch1.1.0 实现 注意：AlexNet，in_img_size >=64 输入图片矩阵的大小要大于等于64 # coding:utf-8 import torch.nn as nn import torch class alex_net(nn.Module) 阅读全文
posted @ 2020-04-07 16:32 算法浪客 阅读(1761) 评论(2) 推荐(0) 编辑
pytorch实现vgg19 训练自定义分类图片
摘要：1、vgg19模型——pytorch 版本= 1.1.0 实现 # coding:utf-8 import torch.nn as nn import torch class vgg19_Net(nn.Module): def __init__(self,in_img_rgb=3,in_img_si 阅读全文
posted @ 2020-04-07 13:42 算法浪客 阅读(4244) 评论(0) 推荐(0) 编辑
奉献pytorch 搭建 CNN 卷积神经网络训练图像识别的模型，配合numpy 和matplotlib 一起使用调用 cuda GPU进行加速训练
摘要：1、Torch构建简单的模型 # coding:utf-8 import torch class Net(torch.nn.Module): def __init__(self,img_rgb=3,img_size=32,img_class=13): super(Net, self).__init_ 阅读全文
posted @ 2019-08-24 22:52 算法浪客 阅读(775) 评论(0) 推荐(0) 编辑
python pytorch numpy DNN 线性回归模型
摘要：1、直接奉献代码，后期有入门更新，之前一直在学的是TensorFlow， import torch from torch.autograd import Variable import torch.nn.functional as F import matplotlib.pyplot as plt 阅读全文
posted @ 2019-08-16 22:56 算法浪客 阅读(2239) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示