zoukankan      html  css  js  c++  java
  • pytorch入门与实践-2.2-CIFAR10分类网络

    1--数据载入

    |----流程: DataSet->DataLoader->调用DataLoader

    |----DataLoader迭代器读不到数据,无报错,一直卡住的显现:

      DataLoader的num_worker设置为0 解决,这个多线程读取数据不知道为什么会出现这个bug?

    2--定义网络

    |----与之前的一致

    3--损失函数和优化器

    |----损失函数: 交叉熵

    |----优化器:SGD (?)

    4--网络训练

    |----读入数据

    |----输入网络

    |----获得结果,计算loss

    |----清空梯度

    |----反向传播

    |--训练结束

    5--使用cuda加速

    |----先将网络载入显卡 net.cuda(),再将运算的数据也载如cuda, x.cuda()

    |----注意处:但是网络传播时输入的还应该是 Variable对象,只是底层利用了cuda对象实现

    |----问题:?虽然占用了部分GPU但是没感觉有啥速度上的变化? batch增大后会卡死(batch==64时巨卡)?

    |---原因分析: 也许是因为每个batch单独载入GPU的问题? 如何在开始时一次性载入 ? 如何不重复载入?

    6--第二章总结

    |----pytorch 三元素 Tensor,Variable,NN

  • 相关阅读:
    vue登录
    登录vue
    FoolWeb Logo
    无限极下拉列表管理
    qtip2 扩展.net mvc 默认验证样式
    编辑器上传提交不能过.net 安全验证的解决方案
    FoolWeb 各层代码实例
    iis7 上传大文件那点事
    js制作 电子杂志(附源代码)
    FoolWeb (二) 数据库设计
  • 原文地址:https://www.cnblogs.com/CXianRen/p/10443344.html
Copyright © 2011-2022 走看看