![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
torch
千羽QY
这个作者很懒,什么都没留下…
展开
-
pytorch 通过load_state_dict加载权重
1、如果预训练网络与当前网络一模一样net = model(input_channels=input_channels, angle_cls=args.angle_bins)# 加载预训练模型pretrained_dict = torch.load("预训练权重路径")net.load_state_dict(pretrained_dict) # True:完全吻合,False:只加载键值相同的参数,其他加载默认值。2、如果预训练网络和当前网络中有一部分网络层的名称不同如,预训练网络中有一层原创 2022-01-20 11:18:53 · 1623 阅读 · 0 评论 -
ninja: build stopped: subcommand failed.
跑DANet时,当运行至以下代码时,出现如题错误;import osimport torchfrom torch.utils.cpp_extension import loadcwd = os.path.dirname(os.path.realpath(__file__))cpu_path = os.path.join(cwd, 'cpu')gpu_path = os.path.join(cwd, 'gpu')cpu = load('enclib_cpu', [ os.pa原创 2022-01-12 19:51:03 · 3296 阅读 · 1 评论 -
torch——通过复制扩展维度
label.shape=(n, 28, 28)label = torch.repeat_interleave(label.unsqueeze(dim=1), repeats=5, dim=1) # (n, 8, 56, 56)label.shape=(n, 5, 28, 28)原创 2020-08-21 15:44:56 · 5177 阅读 · 0 评论