
PyTorch
爱吃文字的鲸鱼
这个作者很懒,什么都没留下…
展开
-
torch.unsqueeze
in:import numpy as npimport torcha=torch.Tensor([1,2,3,4])print(a)print(a.size())b=torch.unsqueeze(a,0)print(b)print(b.size())c=torch.unsqueeze(a,1)print(c)print(c.size())out:tenso...转载 2019-03-31 10:28:55 · 378 阅读 · 0 评论 -
mysql-task2
1.#作业#创建如下所示的 courses 表 ,有: student (学生) 和 class (课程)。例如,表:+---------+------------+| student | class |+---------+------------+| A | Math || B | English || C...原创 2019-04-04 20:31:08 · 90 阅读 · 0 评论 -
colab从谷歌云端硬盘中读入模块
1.输入如下代码from google.colab import drivedrive.mount('/content/drive')结果:2. 查看模块所在文件夹下的内容和导入模块!ls "/content/drive/My Drive/python/MNIST"!cp "/content/drive/My Drive/python/MNIST/net.py" "...原创 2019-04-06 15:03:12 · 3135 阅读 · 1 评论 -
pytorch模型的保存及加载
pytorch模型的保存及加载主要用到下面几行代码#模型的保存,在模型训练完以后torch.save(model.state_dict(),'./model.pt')#模型的加载model1=simpleNet(28*28,10,300,100)#输入原本模型所需的参数,定义的网络的名称为simpleNet,如果是在同一个程序中运行,直接调用就可以,如果不是同一个程序中,那么...原创 2019-04-08 21:16:44 · 480 阅读 · 0 评论 -
pytorch遇到的问题1
问题一RuntimeError:Given groups=1,weight of size[16,3,3,3],expected input[1,384,384,3] to have 3 channels,but got 384 channels instead.我写的网络的nn.Conv2d(3,16,3,1,padding=1),要求输入通道为3,输出是16,网络对于四个参数的认识是第一...原创 2019-04-22 11:27:02 · 5130 阅读 · 2 评论 -
Jupyter Notebook运行包含外部参数的Python程序报错
转载一篇:https://blog.youkuaiyun.com/ken_for_learning/article/details/89675904转载 2019-05-25 10:06:13 · 979 阅读 · 0 评论