- 博客(3)
- 收藏
- 关注
RSS订阅原创 2021-01-15
我想问问 图像修复在训练的时候用的是整个不规则掩码进行训练 然后测试时根据不同的mask ratio分别进行测试验证效果 还是在训练的时候就是根据不同mask ratio分别进行训练 测试就用相同mask ratio进行测试 ...
2021-01-15 15:20:57
82
1
原创 2020-11-15
使用了selfattention 加入了别人的模型框架中 发现cuda out of memory，消耗的内存从2g直接跳到了10g，源代码是文章大神的没动过 。 class Self_Attn(nn.Module): “”" Self attention Layer""" def init(self,in_dim,activation,sn=False): super(Self_Attn,self).__init__() self.channel_in = in_dim self.
2020-11-15 15:15:17
122
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人