Pytorch
西檬饭
这个作者很懒,什么都没留下…
展开
-
使用单机单机多卡的训练模型时复信号输入问题
项目场景:使用单机多卡的模式训练模型问题描述torch.complex类型数据为模型输入,单卡训练正常,多卡训练报错。原因分析:单机多卡训练过程:解决方案:提示:这里填写该问题的具体解决方案:例如:新建一个 Message 对象,并将读取到的数据存入 Message,然后 mHandler.obtainMessage(READ_DATA, bytes, -1, buffer).sendToTarget();换成 mHandler.sendMessage()。...原创 2022-03-20 15:42:16 · 362 阅读 · 0 评论 -
AttributeError: module ‘torch.onnx‘ has no attribute ‘set_training‘
问题描述import torchimport tensorwatch as twresnet18 = models.resnet18(pretrained=True)tw.draw_model(resnet18, (1, 3, 224, 224))使用tensorwatch展示网络结构时报错。问题原因当前环境中Pytorch版本为1.8。高于1.6的版本中,模块’torch.onnx’去掉了 'set_training’这个属性。解决办法将Pytorch降到1.6以下。Conda 安原创 2021-06-23 10:23:13 · 3060 阅读 · 7 评论 -
Pytorch 轮子1 基本张量的处理
常用张量的定义全零矩阵、全一矩阵和对角矩阵:>>> torch.zeros([2, 4], dtype=torch.int32)tensor([[ 0, 0, 0, 0], [ 0, 0, 0, 0]], dtype=torch.int32)>>> cuda0 = torch.device('cuda:0')>>> torch.ones([2, 4], dtype=torch.float64, device=cuda原创 2021-06-20 18:03:40 · 762 阅读 · 0 评论