![](https://img-blog.csdnimg.cn/20201014180756724.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
魇餍
你若盛开,蝴蝶自来
展开
-
随机种子固定
import torchimport numpy as npimport randomseed = 0torch.manual_seed(seed)torch.cuda.manual_seed_all(seed)np.random.seed(seed)random.seed(seed)原创 2020-12-26 11:38:55 · 638 阅读 · 0 评论 -
pytorch 创建网络的8种方法
import torch.nn as nnfrom torch.nn import Conv2dfrom collections import OrderedDictclass net1(nn.Module): def __init__(self): super(net1, self).__init__() self.conv1 = Conv2d(...原创 2018-12-04 23:25:03 · 1833 阅读 · 0 评论 -
pytorch 自定义数据库
from torch.utils import dataimport osfrom PIL import Imagefrom torchvision import transforms as Ttransform = T.Compose([ T.Resize(224), # 缩放图片,保持长宽比不变,最短边为224px T.CenterCrop(224), # 从图...原创 2018-12-05 12:33:45 · 438 阅读 · 0 评论 -
The nomber of SenceFlow data
FlyingThing3D train: 22390test: 4370Monkaa train: 8864Drive train: 4400原创 2018-12-21 14:44:13 · 208 阅读 · 0 评论 -
ubuntu安装显卡驱动
方法一:ppa源安装驱动sudo add-apt-repository ppa:graphics-drivers/ppasudo apt-get updatesudo apt-get install nvidia-(具体版本)sudo apt-get install mesa-common-devsudo apt-get install freeglut3-dev方法二:官网下载.ru...原创 2019-03-28 17:42:13 · 2795 阅读 · 0 评论