摘要: def try_gpu(i=0): #@save """如果存在,则返回gpu(i),否则返回cpu()""" if torch.cuda.device_count() >= i + 1: return torch.device(f'cuda:{i}') return torch.device('c 阅读全文
posted @ 2022-01-29 11:57 寻找烟花 阅读(51) 评论(0) 推荐(0) 编辑
摘要: class test(nn.Module): def __init__(self,name): super(test,self).__init__() self.name = name def forward(self,x): return str(x)+self.name net = test(" 阅读全文
posted @ 2021-11-12 09:44 寻找烟花 阅读(280) 评论(0) 推荐(0) 编辑
摘要: 阅读全文
posted @ 2021-11-09 16:02 寻找烟花 阅读(160) 评论(0) 推荐(0) 编辑