pytorch入门与实践-2.2-CIFAR10分类网络

1--数据载入

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

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

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

2--定义网络

|----与之前的一致

3--损失函数和优化器

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

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

扫描二维码关注公众号,回复: 5342052 查看本文章

4--网络训练

|----读入数据

|----输入网络

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

|----清空梯度

|----反向传播

|--训练结束

猜你喜欢

转载自www.cnblogs.com/CXianRen/p/10443344.html