摘要: 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)