我想知道是否有办法用我自己的值初始化 pytorch 卷积过滤器。
例如,我有一个元组[-0.8423, 0.3778][-3.1070, -2.6518]
我想用这些值初始化 2X2 过滤器,我该怎么做?我查找了一些答案,但他们大多使用火炬正态分布和其他答案。没有什么是我不能指定的。
让我知道。
您可以直接为权重分配值:
conv = nn.Conv2d(1, 1, kernel_size=2)
with torch.no_grad():
conv.weight.data = torch.tensor([[-0.8423, 0.3778],[-3.1070, -2.6518]]) # you might need to play a bit with the dimensionality of this tensor
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)