码迷,mamicode.com
首页 >  
搜索关键字:pytorch    ( 1103个结果
resnet mindspore
resnet50 pytorch版本实现 torchvision.models.resnet import torch.nn as nn import math import torch.utils.model_zoo as model_zoo __all__ = ['ResNet', 'resne ...
分类:Web程序   时间:2021-06-10 18:31:55    阅读次数:0
Yolov5模型部署与训练
模型部署 安装 保证CUDA和pytorch安装好的基础上,将YOLOv5的源码拷到本地。 $ git clone https://github.com/ultralytics/yolov5.git 在Anaconda的shell里面,安装官方给好的配置文件requirements.txt。 $ p ...
分类:其他好文   时间:2021-06-02 18:58:40    阅读次数:0
为什么用tensor不用array?
在DL的科研和工程中,对数据的操作是一个重要的环节,其中最常用的工具是Numpy和Pytorch。Numpy一个强大的数据操作的工具,但是它不能在GPU上运行,只有将Numpy中的ndarray转换成tensor, 才能在GPU上运行。所以我们在必要的时候,需要对ndarray和tensor进行操作 ...
分类:其他好文   时间:2021-06-02 18:31:44    阅读次数:0
3.2使用PyTorch搭建AlexNet并训练花分类数据集
恢复内容开始 1、搭建AlexNet网络 2、如何使用自己的数据集使用网络 —————————————————————————————————————————————————————— 1、搭建AlexNet网络 网络结构: 定义网络类: 定义初始化函数:定义网络在正向传播中所需要使用的层结构。 f ...
分类:Web程序   时间:2021-06-02 17:21:33    阅读次数:0
DCGAN的原理及Pytorch实现
follow this video: https://www.youtube.com/watch?v=IZtv9s_Wx9I paper: https://arxiv.org/pdf/1511.06434.pdf 与Simple GAN相比,loss部分是一样的,主要的不同在于结构,从全连接改成卷积 ...
分类:其他好文   时间:2021-05-24 14:04:47    阅读次数:0
Simple GAN的原理及Pytorch实现
follow this video: https://www.youtube.com/watch?v=OljTVUVzPpM paper: https://papers.nips.cc/paper/2014/file/5ca3e9b122f61f8f06494c97b1afccf3-Paper.pd ...
分类:其他好文   时间:2021-05-24 11:57:23    阅读次数:0
好的github链接
1.PyTorch-SE-ResNet https://github.com/StickCui/PyTorch-SE-ResNet 2.resnet https://github.com/lxztju/pytorch_classification/blob/master/models/vision/ ...
分类:其他好文   时间:2021-05-24 01:55:28    阅读次数:0
CUDA error: device-side assert triggered
运行时出现下面的错误: C:/cb/pytorch_1000000000000/work/aten/src/ATen/native/cuda/IndexKernel.cu:142: block: [0,0,0], thread: [17,0,0] Assertion `index >= -sizes ...
分类:其他好文   时间:2021-05-03 12:58:43    阅读次数:0
关于pytorch不能在jupyter notebook中使用的问题
保持技术的好奇心,是一件不错的事情。 pytorch可以在终端导入,但是就是不能再jupyter notebook中导入,网上很多人遇到这个问题, 但是都解决不了我的问题。 无意间看得到安装nb-conda,于是跟着安装了,但是不起作用。 临了准备放弃的时候,想知道nb-conda是什么东西,于是找 ...
分类:其他好文   时间:2021-04-28 11:54:03    阅读次数:0
Pytorch系列:(三)模型构建
nn.Module 函数详解 nn.Module是所有网络模型结构的基类,无论是pytorch自带的模型,还是要自定义模型,都需要继承这个类。这个模块包含了很多子模块,如下所示,_parameters存放的是模型的参数,_buffers也存放的是模型的参数,但是是那些不需要更新的参数。带hook的都 ...
分类:其他好文   时间:2021-04-28 11:52:07    阅读次数:0
1103条   上一页 1 2 3 4 ... 111 下一页
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!