Pytorch
midori_27
nlp
展开
-
Pytorch 使用多GPU训练,有用
model 先把model都定义好,再统一放到GPU上,不然可能会报错 不要动optimizer # 多GPU if len(conf.device_ids)>1: self.model = torch.nn.DataParallel(self.model, device_ids=conf.device_ids).cuda() ...原创 2019-01-08 15:26:12 · 5190 阅读 · 4 评论 -
Pytorch 快速入门 :从tensorflow转Pytorch的笔记(gather的用法,待补充...)
从tensorflow那边踩坑后过来,对比一下二者的不同: PyTorch vs TensorFlow,哪个更适合你 为什么要转Pytorch? 更加灵活(可以去看别人用tensorflow实现,因为很流行,然后方便地用torch去实现自己的想法),支持Python化(也就是说基本可以当numpy使用) 速度更快 代码简洁容易上手(0.4版本后没有Variable之类的定义,只有一个tensor...原创 2019-01-04 18:19:30 · 4540 阅读 · 2 评论