
pytorch学习
我的心永远是笑笑哒
这个作者很懒,什么都没留下…
展开
-
center loss pytorch实现总结
center loss pytorch实现总结原创 2022-09-18 15:12:16 · 1043 阅读 · 0 评论 -
pytorch ignite使用
pytorch ignite使用原创 2022-09-07 12:59:30 · 185 阅读 · 0 评论 -
pytorch_lightning 使用教程
pytorch_lightning使用教程原创 2022-09-07 00:09:57 · 2553 阅读 · 0 评论 -
python将列表元素解释为函数参数
python将列表元素解释为函数参数原创 2022-08-31 19:19:21 · 610 阅读 · 0 评论 -
tensorflow与pytorch转换总结
tensorfow与pytorch进行转换原创 2022-08-30 23:46:22 · 1848 阅读 · 3 评论 -
cuda out of memory (epoch=120)时发生
cuda out of memory (epoch=120)时发生原创 2022-08-17 13:26:11 · 361 阅读 · 0 评论 -
cosine 相似度计算总结
cosine相似度计算总结原创 2022-06-10 19:38:49 · 2092 阅读 · 1 评论 -
CPU上keras和pytorch训练速度比较(问题已解决)
我测试发现,keras在比较简单的网络上远比pytorch跑的要快的多,目前大致的猜测原因在读取数据部分,但是我还不知道怎么pytorch这个数据的读取速度,为什么都没有大佬发现这个问题呢?mnist数据集读取import osos.environ["CUDA_VISIBLE_DEVICES"] = "0"import torchfrom torchvision import datasets, transformsimport matplotlib.pyplot as pltimport t原创 2022-05-07 06:17:56 · 1110 阅读 · 2 评论 -
torch.mannul_seed()使用和踩坑总结
import torch torch.manual_seed(0)print(torch.randn(1, 2))print(torch.randn(1, 2))print("======================") # torch.manual_seed()控制torch中随机数的生成,并不控制numpy和random中torch.manual_seed(0)print(torch.randn(1, 2))print(torch.randn(1, 2))然后就有一个好玩的事原创 2022-05-01 16:17:44 · 1835 阅读 · 0 评论 -
pytorch复现结果的坑--集合使用
对于pytorch的结果复现,对于普通的设置使用下面的函数即可参考https://stackoverflow.com/questions/60386708/random-seed-for-order-of-elements-in-pythons-set-to-list-conversionpytorch复现结果def seed_torch(seed=42): random.seed(seed) os.environ['PYTHONHASHSEED'] = str(seed)原创 2022-04-27 20:05:58 · 535 阅读 · 2 评论 -
tensorflow和pytorch设置和查看权重
pytorchimport numpy as npimport torch.nn as nnimport randomimport os import torchimport seaborn as snsimport matplotlib.pyplot as pltdef seed_torch(seed=42): random.seed(seed) os.environ['PYTHONHASHSEED'] = '0' np.random.seed(seed)原创 2022-04-13 09:51:19 · 2262 阅读 · 0 评论 -
tensorlflow和pytorch打印模型参数
tensorflowimport tensorflow.keras.backend as Kfrom tensorflow.keras.callbacks import TensorBoard, ModelCheckpoint, EarlyStopping, ReduceLROnPlateau,Historyfrom tensorflow.keras.layers import Dense, Input,Layer,InputSpecfrom tensorflow.keras.models impo原创 2022-04-03 23:10:16 · 1986 阅读 · 0 评论 -
pytorch earlystopping使用
earlystopping实现https://github.com/Bjarten/early-stopping-pytorch自己实现LinearRegression# 我通过一个线性回归(或者一个自编码器)来联系使用earlystopimport torchimport numpy as npimport torch.nn as nn#X1=np.linspace(-10,10,100)#X2=np.linspace(-20,20,100)#X3=np.linspace(0,40,1原创 2022-04-02 11:24:15 · 2720 阅读 · 0 评论 -
KL散度损失学习
下面进行代码验证单个样本import torch.nn.functional as F# this is the same example in wikiP = F.softmax(torch.randn(1,4),dim=1)Q = F.softmax(torch.randn(1,4),dim=1)print((P * (P / Q).log()).sum()) # # tensor(0.0863), 10.2 µs ± 508print(F.kl_div(Q.log(), P, r原创 2022-03-30 22:49:27 · 1453 阅读 · 0 评论 -
pytorch 卷积自编码器
import torchimport torchvisionfrom torch import nnfrom torch.autograd import Variablefrom torch.utils.data import DataLoaderfrom torchvision import transformsfrom torchvision.utils import save_imagefrom torchvision.datasets import MNISTimport osba原创 2022-03-25 08:58:33 · 5990 阅读 · 2 评论 -
nn.embedding()测试
https://zhuanlan.zhihu.com/p/361431647https://zhuanlan.zhihu.com/p/345450458import torchfrom torch import nnembedding = nn.Embedding(5, 4) # 假定字典中只有5个词,词向量维度为4word = [[1, 2, 3], [2, 3, 4]] # 每个数字代表一个词,例如 {'!':0,'how':1, 'are':2, 'you':3, 'ok原创 2022-03-19 16:57:49 · 1430 阅读 · 0 评论 -
pytorch_CPU_GPU时间测试
参考https://zhuanlan.zhihu.com/p/35434175https://www.kaggle.com/scottclowe/testing-gpu-enabled-notebooks-mnist-pytorch#%%time import torchimport torch.nn as nnimport torch.nn.functional as Fimport numpy as npimport matplotlib.pyplot as pltfrom torch原创 2022-01-16 20:02:14 · 1403 阅读 · 0 评论 -
torch.nn.L1Loss
import torch.nn as nnimport torchloss = nn.L1Loss()a=torch.FloatTensor([1,2,3])b=torch.FloatTensor([1.1,2.2,2.9]) output = loss(a, b)print(output)c=torch.FloatTensor([[1,3],[2,3],[1,1]])d=torch.FloatTensor([[2,3.2],[2.1,3.2],[1.1,1.1]])out=loss(c原创 2021-08-29 02:15:38 · 824 阅读 · 0 评论 -
usps数据集处理
kaggle uspsimport numpy as np # linear algebraimport pandas as pd # data processing, CSV file I/O (e.g. pd.read_csv)import os#print(os.listdir("../input"))import h5py from functools import reducedef hdf5(path, data_key = "data", target_key = "targe原创 2021-04-25 21:51:43 · 4134 阅读 · 0 评论