- 博客(6)
- 资源 (6)
- 收藏
- 关注
RSS订阅转载 pytorch 与tensorflow 的条件判断应用
https://fgc.stpi.narl.org.tw/activity/videoDetail/4b1141305d9cd231015d9d0992ef0030tensorflow中使用tf.cond()实现条件判断，pytorch直接使用if判断
2018-12-26 15:32:10
1403
转载 pytorch 类型不匹配问题
https://blog.csdn.net/hustchenze/article/details/79154139这篇博客写的比较全，特此记录，尤其是 a1.type_as(a2)
2018-12-24 15:26:32
274
原创 gpu 显存占用 与 volatile gpu-util （gpu利用率）
1. https://github.com/tensorflow/tensorflow/issues/5432. http://caffecn.cn/?/question/1861 gpu占用高，不表示利用率高，tensorflow不设置gpu占用多少的情况下，会默认占满gpu 要是需要同时跑几个gpu程序，可以设置tensorflow的最大gpu显存占用，不要设置根据需要增加...
2018-12-07 11:10:51
42306
转载 git clone tensorflow
报错：git@github.com: Permission denied (publickey).fatal: Could not read from remote repository.Please make sure you have the correct access rightsand the repository exists. 配置SSH key，并添加到githu...
2018-12-07 10:31:37
645
转载 pytorch cpu调用gpu训练模型
参考：https://blog.csdn.net/dcrmg/article/details/79503978 model = torch.load('../model/model-new4.pkl', map_location=lambda storage, loc: storage)
2018-12-05 12:01:27
1600
原创 pytorch cuda.FloatTensor->FloatTensor
错误类型：RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor） 定义残差块时定义在model的外面，在使用gpu进行训练的时候，残差块的参数是torch.FloatTensor类型，虽然使用了model.cuda(),但是只对model里面的参数在g...
2018-12-05 10:57:11
2914
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人