logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

pytorch学习(十六)to(device) 和.cuda()的用法

看pytorch官方给的代码,to(device)的时候可以指定GPU设备号。二者没啥太大区别,都可以用。cuda = torch.device('cuda')# Default CUDA devicecuda0 = torch.device('cuda:0')cuda2 = torch.device('cuda:2')# GPU 2 (these are 0-indexed)x = torch.

#深度学习
nn.Sequential(*layers)中*的用法及错误TypeError: list is not a Module subclass

这里主要涉及一些nn.Sequential()的用法,nn.Sequential()是一个顺序容器,将神经网络的相关操作进行封装。1. nn.Sequential()容器定义从nn.Sequential()的定义来看,输入要么是orderdict,要么是一系列的模型,遇到list,必须用*号进行转化,否则会报错 TypeError: list is not a Module subclass2.

#神经网络#pytorch
Android Studio代码首次提交到github上

直到开始做大项目,代码量庞大的时候才意识到版本管理的重要性,每当发现自己写的有问题的时候,就想回溯到上一个版本,结果并不记得自己改动了哪里改动了多少。所以写此篇文章记录流程。1.准备工作(1)安装git(2)github上注册账号2.步骤(1)在AS上配置git打开路径File->Settings->Version Control->Git在Pat...

#git
到底了