摘要:
class conv_try(nn.Module): def __init__(self): super(conv_try, self).__init__() # self.conv1 = nn.Conv1d(1, 32, kernel_size=3, stride=2, padding=1) # 阅读全文
posted @ 2021-11-08 12:27
tanyayangyang
阅读(1172)
评论(0)
推荐(0)
摘要:
有numpy和torch两种类型的数据 z = mean.clone().detach() + eps * torch.exp(logstd) 源代码这边报错了,修改如下 eps = eps.cuda() z = mean.cuda()+ eps * torch.exp(logstd).cuda() 阅读全文
posted @ 2021-11-08 09:49
tanyayangyang
阅读(2074)
评论(0)
推荐(0)

浙公网安备 33010602011771号