pytorch
_Cade_
QQ:1004573547
展开
-
python 版本 torchnet 简单使用文档
最近复现prototypical net发现作者源代码使用了torchnet中的meter,还写了一个engine。所以我用一天时间看了下这个库。其实不用这个也是完全没有任何问题的。它只是方便复用的一个框架torchnet是用于 torch 的代码复用和模块化编程的框架文档:https://tnt.readthedocs.io/en/latest/github地址:https:...原创 2019-05-16 20:05:46 · 7600 阅读 · 1 评论 -
pytorch 踩坑记录
其他坑一些别人踩过的坑,知乎上的回答1.Broadcast function not implemented for CPU tensors 这是因为model不在gpu上所致。model.to(device)。DataParallel会对模型参数所在的gpu位置进行检查,见源码 DataParallel是每次forward时对模型进行broadcast,当模型不在第一个G...原创 2019-06-16 10:41:23 · 17193 阅读 · 0 评论