码迷,mamicode.com
首页 >  
搜索关键字:caffe2 PyTorch    ( 1125个结果
nn.CrossEntropyLoss()->nn.SoftmaxCrossEntropyWithLogits
问题描述: 在pytorch中,利用nn.CrossEntropyLoss()求取的loss值是一个tensor标量,但是利用mindspore中nn.SoftmaxCrossEntropyWithLogits()求取的loss值是一个矢量(因为batct_size是32,所以求得的loss也是一个 ...
分类:其他好文   时间:2021-06-10 18:33:56    阅读次数:0
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
1125条   上一页 1 2 3 4 ... 113 下一页
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!