pytorch
渣渣琪
菜鸟成长记
展开
-
使用 pip install torchvision 报错
使用 pip install torchvision 报错解决办法: 换命令: conda install torchvision 或者 conda install torchvision -c pytorch 亲测第一个成功,第二个可能不行原创 2020-06-30 10:20:54 · 2223 阅读 · 2 评论 -
Pytorch快速搭建神经网络
(一)init和forward 每一个torch模块当中都有__init__()和forward() __init__()里面包含搭建神经网络那些层所需要的信息; forward()包含神经网络前向传递的内容;比如init里面的那些层的组合就是由forward来完成; (二)要继承父类的__init__() (三)搭建层 其中对于hidden:n_features是输入;n_h...原创 2019-08-01 13:55:45 · 368 阅读 · 0 评论