You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

PyTorch_quick_intro.ipynb 64 kB

6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
6 years ago
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532
  1. {
  2. "cells": [
  3. {
  4. "cell_type": "markdown",
  5. "metadata": {},
  6. "source": [
  7. "# PyTorch快速入门\n",
  8. "\n",
  9. "PyTorch的简洁设计使得它入门很简单,在深入介绍PyTorch之前,本节将先介绍一些PyTorch的基础知识,使得读者能够对PyTorch有一个大致的了解,并能够用PyTorch搭建一个简单的神经网络。部分内容读者可能暂时不太理解,可先不予以深究,后续的课程将会对此进行深入讲解。\n",
  10. "\n",
  11. "本节内容参考了PyTorch官方教程[^1]并做了相应的增删修改,使得内容更贴合新版本的PyTorch接口,同时也更适合新手快速入门。另外本书需要读者先掌握基础的Numpy使用,其他相关知识推荐读者参考CS231n的教程[^2]。\n",
  12. "\n",
  13. "[^1]: http://pytorch.org/tutorials/beginner/deep_learning_60min_blitz.html\n",
  14. "[^2]: http://cs231n.github.io/python-numpy-tutorial/"
  15. ]
  16. },
  17. {
  18. "cell_type": "markdown",
  19. "metadata": {},
  20. "source": [
  21. "### Tensor\n",
  22. "\n",
  23. "Tensor是PyTorch中重要的数据结构,可认为是一个高维数组。它可以是一个数(标量)、一维数组(向量)、二维数组(矩阵)以及更高维的数组。Tensor和Numpy的ndarrays类似,但Tensor可以使用GPU进行加速。Tensor的使用和Numpy及Matlab的接口十分相似,下面通过几个例子来看看Tensor的基本使用。"
  24. ]
  25. },
  26. {
  27. "cell_type": "code",
  28. "execution_count": 1,
  29. "metadata": {},
  30. "outputs": [],
  31. "source": [
  32. "from __future__ import print_function\n",
  33. "import torch as t"
  34. ]
  35. },
  36. {
  37. "cell_type": "code",
  38. "execution_count": 2,
  39. "metadata": {},
  40. "outputs": [
  41. {
  42. "data": {
  43. "text/plain": [
  44. "\n",
  45. "1.00000e-07 *\n",
  46. " 0.0000 0.0000 5.3571\n",
  47. " 0.0000 0.0000 0.0000\n",
  48. " 0.0000 0.0000 0.0000\n",
  49. " 0.0000 5.4822 0.0000\n",
  50. " 5.4823 0.0000 5.4823\n",
  51. "[torch.FloatTensor of size 5x3]"
  52. ]
  53. },
  54. "execution_count": 2,
  55. "metadata": {},
  56. "output_type": "execute_result"
  57. }
  58. ],
  59. "source": [
  60. "# 构建 5x3 矩阵,只是分配了空间,未初始化\n",
  61. "x = t.Tensor(5, 3) \n",
  62. "x"
  63. ]
  64. },
  65. {
  66. "cell_type": "code",
  67. "execution_count": 3,
  68. "metadata": {},
  69. "outputs": [
  70. {
  71. "data": {
  72. "text/plain": [
  73. "\n",
  74. " 0.3673 0.2522 0.3553\n",
  75. " 0.0070 0.7138 0.0463\n",
  76. " 0.6198 0.6019 0.3752\n",
  77. " 0.4755 0.3675 0.3032\n",
  78. " 0.5824 0.5104 0.5759\n",
  79. "[torch.FloatTensor of size 5x3]"
  80. ]
  81. },
  82. "execution_count": 3,
  83. "metadata": {},
  84. "output_type": "execute_result"
  85. }
  86. ],
  87. "source": [
  88. "# 使用[0,1]均匀分布随机初始化二维数组\n",
  89. "x = t.rand(5, 3) \n",
  90. "x"
  91. ]
  92. },
  93. {
  94. "cell_type": "code",
  95. "execution_count": 4,
  96. "metadata": {},
  97. "outputs": [
  98. {
  99. "name": "stdout",
  100. "output_type": "stream",
  101. "text": [
  102. "torch.Size([5, 3])\n"
  103. ]
  104. },
  105. {
  106. "data": {
  107. "text/plain": [
  108. "(3, 3)"
  109. ]
  110. },
  111. "execution_count": 4,
  112. "metadata": {},
  113. "output_type": "execute_result"
  114. }
  115. ],
  116. "source": [
  117. "print(x.size()) # 查看x的形状\n",
  118. "x.size()[1], x.size(1) # 查看列的个数, 两种写法等价"
  119. ]
  120. },
  121. {
  122. "cell_type": "markdown",
  123. "metadata": {},
  124. "source": [
  125. "`torch.Size` 是tuple对象的子类,因此它支持tuple的所有操作,如x.size()[0]"
  126. ]
  127. },
  128. {
  129. "cell_type": "code",
  130. "execution_count": 5,
  131. "metadata": {},
  132. "outputs": [
  133. {
  134. "data": {
  135. "text/plain": [
  136. "\n",
  137. " 0.4063 0.7378 1.2411\n",
  138. " 0.0687 0.7725 0.0634\n",
  139. " 1.1016 1.4291 0.7324\n",
  140. " 0.7604 1.2880 0.4597\n",
  141. " 0.6020 1.0124 1.0185\n",
  142. "[torch.FloatTensor of size 5x3]"
  143. ]
  144. },
  145. "execution_count": 5,
  146. "metadata": {},
  147. "output_type": "execute_result"
  148. }
  149. ],
  150. "source": [
  151. "y = t.rand(5, 3)\n",
  152. "# 加法的第一种写法\n",
  153. "x + y"
  154. ]
  155. },
  156. {
  157. "cell_type": "code",
  158. "execution_count": 6,
  159. "metadata": {},
  160. "outputs": [
  161. {
  162. "data": {
  163. "text/plain": [
  164. "\n",
  165. " 0.4063 0.7378 1.2411\n",
  166. " 0.0687 0.7725 0.0634\n",
  167. " 1.1016 1.4291 0.7324\n",
  168. " 0.7604 1.2880 0.4597\n",
  169. " 0.6020 1.0124 1.0185\n",
  170. "[torch.FloatTensor of size 5x3]"
  171. ]
  172. },
  173. "execution_count": 6,
  174. "metadata": {},
  175. "output_type": "execute_result"
  176. }
  177. ],
  178. "source": [
  179. "# 加法的第二种写法\n",
  180. "t.add(x, y)"
  181. ]
  182. },
  183. {
  184. "cell_type": "code",
  185. "execution_count": 7,
  186. "metadata": {},
  187. "outputs": [
  188. {
  189. "data": {
  190. "text/plain": [
  191. "\n",
  192. " 0.4063 0.7378 1.2411\n",
  193. " 0.0687 0.7725 0.0634\n",
  194. " 1.1016 1.4291 0.7324\n",
  195. " 0.7604 1.2880 0.4597\n",
  196. " 0.6020 1.0124 1.0185\n",
  197. "[torch.FloatTensor of size 5x3]"
  198. ]
  199. },
  200. "execution_count": 7,
  201. "metadata": {},
  202. "output_type": "execute_result"
  203. }
  204. ],
  205. "source": [
  206. "# 加法的第三种写法:指定加法结果的输出目标为result\n",
  207. "result = t.Tensor(5, 3) # 预先分配空间\n",
  208. "t.add(x, y, out=result) # 输入到result\n",
  209. "result"
  210. ]
  211. },
  212. {
  213. "cell_type": "code",
  214. "execution_count": 8,
  215. "metadata": {},
  216. "outputs": [
  217. {
  218. "name": "stdout",
  219. "output_type": "stream",
  220. "text": [
  221. "最初y\n",
  222. "\n",
  223. " 0.0390 0.4856 0.8858\n",
  224. " 0.0617 0.0587 0.0171\n",
  225. " 0.4818 0.8272 0.3572\n",
  226. " 0.2849 0.9205 0.1565\n",
  227. " 0.0196 0.5020 0.4426\n",
  228. "[torch.FloatTensor of size 5x3]\n",
  229. "\n",
  230. "第一种加法,y的结果\n",
  231. "\n",
  232. " 0.0390 0.4856 0.8858\n",
  233. " 0.0617 0.0587 0.0171\n",
  234. " 0.4818 0.8272 0.3572\n",
  235. " 0.2849 0.9205 0.1565\n",
  236. " 0.0196 0.5020 0.4426\n",
  237. "[torch.FloatTensor of size 5x3]\n",
  238. "\n",
  239. "第二种加法,y的结果\n",
  240. "\n",
  241. " 0.4063 0.7378 1.2411\n",
  242. " 0.0687 0.7725 0.0634\n",
  243. " 1.1016 1.4291 0.7324\n",
  244. " 0.7604 1.2880 0.4597\n",
  245. " 0.6020 1.0124 1.0185\n",
  246. "[torch.FloatTensor of size 5x3]\n",
  247. "\n"
  248. ]
  249. }
  250. ],
  251. "source": [
  252. "print('最初y')\n",
  253. "print(y)\n",
  254. "\n",
  255. "print('第一种加法,y的结果')\n",
  256. "y.add(x) # 普通加法,不改变y的内容\n",
  257. "print(y)\n",
  258. "\n",
  259. "print('第二种加法,y的结果')\n",
  260. "y.add_(x) # inplace 加法,y变了\n",
  261. "print(y)"
  262. ]
  263. },
  264. {
  265. "cell_type": "markdown",
  266. "metadata": {},
  267. "source": [
  268. "注意,函数名后面带下划线**`_`** 的函数会修改Tensor本身。例如,`x.add_(y)`和`x.t_()`会改变 `x`,但`x.add(y)`和`x.t()`返回一个新的Tensor, 而`x`不变。"
  269. ]
  270. },
  271. {
  272. "cell_type": "code",
  273. "execution_count": 9,
  274. "metadata": {},
  275. "outputs": [
  276. {
  277. "data": {
  278. "text/plain": [
  279. "\n",
  280. " 0.2522\n",
  281. " 0.7138\n",
  282. " 0.6019\n",
  283. " 0.3675\n",
  284. " 0.5104\n",
  285. "[torch.FloatTensor of size 5]"
  286. ]
  287. },
  288. "execution_count": 9,
  289. "metadata": {},
  290. "output_type": "execute_result"
  291. }
  292. ],
  293. "source": [
  294. "# Tensor的选取操作与Numpy类似\n",
  295. "x[:, 1]"
  296. ]
  297. },
  298. {
  299. "cell_type": "markdown",
  300. "metadata": {},
  301. "source": [
  302. "Tensor还支持很多操作,包括数学运算、线性代数、选择、切片等等,其接口设计与Numpy极为相似。更详细的使用方法,会在第三章系统讲解。\n",
  303. "\n",
  304. "Tensor和Numpy的数组之间的互操作非常容易且快速。对于Tensor不支持的操作,可以先转为Numpy数组处理,之后再转回Tensor。"
  305. ]
  306. },
  307. {
  308. "cell_type": "code",
  309. "execution_count": 10,
  310. "metadata": {},
  311. "outputs": [
  312. {
  313. "data": {
  314. "text/plain": [
  315. "\n",
  316. " 1\n",
  317. " 1\n",
  318. " 1\n",
  319. " 1\n",
  320. " 1\n",
  321. "[torch.FloatTensor of size 5]"
  322. ]
  323. },
  324. "execution_count": 10,
  325. "metadata": {},
  326. "output_type": "execute_result"
  327. }
  328. ],
  329. "source": [
  330. "a = t.ones(5) # 新建一个全1的Tensor\n",
  331. "a"
  332. ]
  333. },
  334. {
  335. "cell_type": "code",
  336. "execution_count": 11,
  337. "metadata": {},
  338. "outputs": [
  339. {
  340. "data": {
  341. "text/plain": [
  342. "array([1., 1., 1., 1., 1.], dtype=float32)"
  343. ]
  344. },
  345. "execution_count": 11,
  346. "metadata": {},
  347. "output_type": "execute_result"
  348. }
  349. ],
  350. "source": [
  351. "b = a.numpy() # Tensor -> Numpy\n",
  352. "b"
  353. ]
  354. },
  355. {
  356. "cell_type": "code",
  357. "execution_count": 12,
  358. "metadata": {},
  359. "outputs": [
  360. {
  361. "name": "stdout",
  362. "output_type": "stream",
  363. "text": [
  364. "[1. 1. 1. 1. 1.]\n",
  365. "\n",
  366. " 1\n",
  367. " 1\n",
  368. " 1\n",
  369. " 1\n",
  370. " 1\n",
  371. "[torch.DoubleTensor of size 5]\n",
  372. "\n"
  373. ]
  374. }
  375. ],
  376. "source": [
  377. "import numpy as np\n",
  378. "a = np.ones(5)\n",
  379. "b = t.from_numpy(a) # Numpy->Tensor\n",
  380. "print(a)\n",
  381. "print(b) "
  382. ]
  383. },
  384. {
  385. "cell_type": "markdown",
  386. "metadata": {},
  387. "source": [
  388. "Tensor和numpy对象共享内存,所以他们之间的转换很快,而且几乎不会消耗什么资源。但这也意味着,如果其中一个变了,另外一个也会随之改变。"
  389. ]
  390. },
  391. {
  392. "cell_type": "code",
  393. "execution_count": 13,
  394. "metadata": {},
  395. "outputs": [
  396. {
  397. "name": "stdout",
  398. "output_type": "stream",
  399. "text": [
  400. "[2. 2. 2. 2. 2.]\n",
  401. "\n",
  402. " 2\n",
  403. " 2\n",
  404. " 2\n",
  405. " 2\n",
  406. " 2\n",
  407. "[torch.DoubleTensor of size 5]\n",
  408. "\n"
  409. ]
  410. }
  411. ],
  412. "source": [
  413. "b.add_(1) # 以`_`结尾的函数会修改自身\n",
  414. "print(a)\n",
  415. "print(b) # Tensor和Numpy共享内存"
  416. ]
  417. },
  418. {
  419. "cell_type": "markdown",
  420. "metadata": {},
  421. "source": [
  422. "Tensor可通过`.cuda` 方法转为GPU的Tensor,从而享受GPU带来的加速运算。"
  423. ]
  424. },
  425. {
  426. "cell_type": "code",
  427. "execution_count": 14,
  428. "metadata": {},
  429. "outputs": [],
  430. "source": [
  431. "# 在不支持CUDA的机器下,下一步不会运行\n",
  432. "if t.cuda.is_available():\n",
  433. " x = x.cuda()\n",
  434. " y = y.cuda()\n",
  435. " x + y"
  436. ]
  437. },
  438. {
  439. "cell_type": "markdown",
  440. "metadata": {},
  441. "source": [
  442. "此处可能发现GPU运算的速度并未提升太多,这是因为x和y太小且运算也较为简单,而且将数据从内存转移到显存还需要花费额外的开销。GPU的优势需在大规模数据和复杂运算下才能体现出来。\n",
  443. "\n",
  444. "### Autograd: 自动微分\n",
  445. "\n",
  446. "深度学习的算法本质上是通过反向传播求导数,而PyTorch的**`Autograd`**模块则实现了此功能。在Tensor上的所有操作,Autograd都能为它们自动提供微分,避免了手动计算导数的复杂过程。\n",
  447. " \n",
  448. "`autograd.Variable`是Autograd中的核心类,它简单封装了Tensor,并支持几乎所有Tensor有的操作。Tensor在被封装为Variable之后,可以调用它的`.backward`实现反向传播,自动计算所有梯度。Variable的数据结构如图2-6所示。\n",
  449. "\n",
  450. "\n",
  451. "![图2-6:Variable的数据结构](imgs/autograd_Variable.svg)\n",
  452. "\n",
  453. "\n",
  454. "Variable主要包含三个属性。\n",
  455. "- `data`:保存Variable所包含的Tensor\n",
  456. "- `grad`:保存`data`对应的梯度,`grad`也是个Variable,而不是Tensor,它和`data`的形状一样。\n",
  457. "- `grad_fn`:指向一个`Function`对象,这个`Function`用来反向传播计算输入的梯度,具体细节会在下一章讲解。"
  458. ]
  459. },
  460. {
  461. "cell_type": "code",
  462. "execution_count": 15,
  463. "metadata": {},
  464. "outputs": [],
  465. "source": [
  466. "from torch.autograd import Variable"
  467. ]
  468. },
  469. {
  470. "cell_type": "code",
  471. "execution_count": 16,
  472. "metadata": {
  473. "scrolled": true
  474. },
  475. "outputs": [
  476. {
  477. "data": {
  478. "text/plain": [
  479. "Variable containing:\n",
  480. " 1 1\n",
  481. " 1 1\n",
  482. "[torch.FloatTensor of size 2x2]"
  483. ]
  484. },
  485. "execution_count": 16,
  486. "metadata": {},
  487. "output_type": "execute_result"
  488. }
  489. ],
  490. "source": [
  491. "# 使用Tensor新建一个Variable\n",
  492. "x = Variable(t.ones(2, 2), requires_grad = True)\n",
  493. "x"
  494. ]
  495. },
  496. {
  497. "cell_type": "code",
  498. "execution_count": 17,
  499. "metadata": {
  500. "scrolled": true
  501. },
  502. "outputs": [
  503. {
  504. "data": {
  505. "text/plain": [
  506. "Variable containing:\n",
  507. " 4\n",
  508. "[torch.FloatTensor of size 1]"
  509. ]
  510. },
  511. "execution_count": 17,
  512. "metadata": {},
  513. "output_type": "execute_result"
  514. }
  515. ],
  516. "source": [
  517. "y = x.sum()\n",
  518. "y"
  519. ]
  520. },
  521. {
  522. "cell_type": "code",
  523. "execution_count": 18,
  524. "metadata": {},
  525. "outputs": [
  526. {
  527. "data": {
  528. "text/plain": [
  529. "<SumBackward0 at 0x7fc14824b860>"
  530. ]
  531. },
  532. "execution_count": 18,
  533. "metadata": {},
  534. "output_type": "execute_result"
  535. }
  536. ],
  537. "source": [
  538. "y.grad_fn"
  539. ]
  540. },
  541. {
  542. "cell_type": "code",
  543. "execution_count": 19,
  544. "metadata": {},
  545. "outputs": [],
  546. "source": [
  547. "y.backward() # 反向传播,计算梯度"
  548. ]
  549. },
  550. {
  551. "cell_type": "code",
  552. "execution_count": 20,
  553. "metadata": {},
  554. "outputs": [
  555. {
  556. "data": {
  557. "text/plain": [
  558. "Variable containing:\n",
  559. " 1 1\n",
  560. " 1 1\n",
  561. "[torch.FloatTensor of size 2x2]"
  562. ]
  563. },
  564. "execution_count": 20,
  565. "metadata": {},
  566. "output_type": "execute_result"
  567. }
  568. ],
  569. "source": [
  570. "# y = x.sum() = (x[0][0] + x[0][1] + x[1][0] + x[1][1])\n",
  571. "# 每个值的梯度都为1\n",
  572. "x.grad "
  573. ]
  574. },
  575. {
  576. "cell_type": "markdown",
  577. "metadata": {},
  578. "source": [
  579. "注意:`grad`在反向传播过程中是累加的(accumulated),**这意味着每一次运行反向传播,梯度都会累加之前的梯度,所以反向传播之前需把梯度清零。**"
  580. ]
  581. },
  582. {
  583. "cell_type": "code",
  584. "execution_count": 21,
  585. "metadata": {},
  586. "outputs": [
  587. {
  588. "data": {
  589. "text/plain": [
  590. "Variable containing:\n",
  591. " 2 2\n",
  592. " 2 2\n",
  593. "[torch.FloatTensor of size 2x2]"
  594. ]
  595. },
  596. "execution_count": 21,
  597. "metadata": {},
  598. "output_type": "execute_result"
  599. }
  600. ],
  601. "source": [
  602. "y.backward()\n",
  603. "x.grad"
  604. ]
  605. },
  606. {
  607. "cell_type": "code",
  608. "execution_count": 22,
  609. "metadata": {
  610. "scrolled": true
  611. },
  612. "outputs": [
  613. {
  614. "data": {
  615. "text/plain": [
  616. "Variable containing:\n",
  617. " 3 3\n",
  618. " 3 3\n",
  619. "[torch.FloatTensor of size 2x2]"
  620. ]
  621. },
  622. "execution_count": 22,
  623. "metadata": {},
  624. "output_type": "execute_result"
  625. }
  626. ],
  627. "source": [
  628. "y.backward()\n",
  629. "x.grad"
  630. ]
  631. },
  632. {
  633. "cell_type": "code",
  634. "execution_count": 23,
  635. "metadata": {},
  636. "outputs": [
  637. {
  638. "data": {
  639. "text/plain": [
  640. "\n",
  641. " 0 0\n",
  642. " 0 0\n",
  643. "[torch.FloatTensor of size 2x2]"
  644. ]
  645. },
  646. "execution_count": 23,
  647. "metadata": {},
  648. "output_type": "execute_result"
  649. }
  650. ],
  651. "source": [
  652. "# 以下划线结束的函数是inplace操作,就像add_\n",
  653. "x.grad.data.zero_()"
  654. ]
  655. },
  656. {
  657. "cell_type": "code",
  658. "execution_count": 24,
  659. "metadata": {},
  660. "outputs": [
  661. {
  662. "data": {
  663. "text/plain": [
  664. "Variable containing:\n",
  665. " 1 1\n",
  666. " 1 1\n",
  667. "[torch.FloatTensor of size 2x2]"
  668. ]
  669. },
  670. "execution_count": 24,
  671. "metadata": {},
  672. "output_type": "execute_result"
  673. }
  674. ],
  675. "source": [
  676. "y.backward()\n",
  677. "x.grad"
  678. ]
  679. },
  680. {
  681. "cell_type": "markdown",
  682. "metadata": {},
  683. "source": [
  684. "Variable和Tensor具有近乎一致的接口,在实际使用中可以无缝切换。"
  685. ]
  686. },
  687. {
  688. "cell_type": "code",
  689. "execution_count": 25,
  690. "metadata": {},
  691. "outputs": [
  692. {
  693. "name": "stdout",
  694. "output_type": "stream",
  695. "text": [
  696. "Variable containing:\n",
  697. " 0.5403 0.5403 0.5403 0.5403 0.5403\n",
  698. " 0.5403 0.5403 0.5403 0.5403 0.5403\n",
  699. " 0.5403 0.5403 0.5403 0.5403 0.5403\n",
  700. " 0.5403 0.5403 0.5403 0.5403 0.5403\n",
  701. "[torch.FloatTensor of size 4x5]\n",
  702. "\n"
  703. ]
  704. },
  705. {
  706. "data": {
  707. "text/plain": [
  708. "\n",
  709. " 0.5403 0.5403 0.5403 0.5403 0.5403\n",
  710. " 0.5403 0.5403 0.5403 0.5403 0.5403\n",
  711. " 0.5403 0.5403 0.5403 0.5403 0.5403\n",
  712. " 0.5403 0.5403 0.5403 0.5403 0.5403\n",
  713. "[torch.FloatTensor of size 4x5]"
  714. ]
  715. },
  716. "execution_count": 25,
  717. "metadata": {},
  718. "output_type": "execute_result"
  719. }
  720. ],
  721. "source": [
  722. "x = Variable(t.ones(4,5))\n",
  723. "y = t.cos(x)\n",
  724. "x_tensor_cos = t.cos(x.data)\n",
  725. "print(y)\n",
  726. "x_tensor_cos"
  727. ]
  728. },
  729. {
  730. "cell_type": "markdown",
  731. "metadata": {},
  732. "source": [
  733. "### 神经网络\n",
  734. "\n",
  735. "Autograd实现了反向传播功能,但是直接用来写深度学习的代码在很多情况下还是稍显复杂,torch.nn是专门为神经网络设计的模块化接口。nn构建于 Autograd之上,可用来定义和运行神经网络。nn.Module是nn中最重要的类,可把它看成是一个网络的封装,包含网络各层定义以及forward方法,调用forward(input)方法,可返回前向传播的结果。下面就以最早的卷积神经网络:LeNet为例,来看看如何用`nn.Module`实现。LeNet的网络结构如图2-7所示。\n",
  736. "\n",
  737. "![图2-7:LeNet网络结构](imgs/nn_lenet.png)\n",
  738. "\n",
  739. "这是一个基础的前向传播(feed-forward)网络: 接收输入,经过层层传递运算,得到输出。\n",
  740. "\n",
  741. "#### 定义网络\n",
  742. "\n",
  743. "定义网络时,需要继承`nn.Module`,并实现它的forward方法,把网络中具有可学习参数的层放在构造函数`__init__`中。如果某一层(如ReLU)不具有可学习的参数,则既可以放在构造函数中,也可以不放,但建议不放在其中,而在forward中使用`nn.functional`代替。"
  744. ]
  745. },
  746. {
  747. "cell_type": "code",
  748. "execution_count": 26,
  749. "metadata": {},
  750. "outputs": [
  751. {
  752. "name": "stdout",
  753. "output_type": "stream",
  754. "text": [
  755. "Net(\n",
  756. " (conv1): Conv2d (1, 6, kernel_size=(5, 5), stride=(1, 1))\n",
  757. " (conv2): Conv2d (6, 16, kernel_size=(5, 5), stride=(1, 1))\n",
  758. " (fc1): Linear(in_features=400, out_features=120)\n",
  759. " (fc2): Linear(in_features=120, out_features=84)\n",
  760. " (fc3): Linear(in_features=84, out_features=10)\n",
  761. ")\n"
  762. ]
  763. }
  764. ],
  765. "source": [
  766. "import torch.nn as nn\n",
  767. "import torch.nn.functional as F\n",
  768. "\n",
  769. "class Net(nn.Module):\n",
  770. " def __init__(self):\n",
  771. " # nn.Module子类的函数必须在构造函数中执行父类的构造函数\n",
  772. " # 下式等价于nn.Module.__init__(self)\n",
  773. " super(Net, self).__init__()\n",
  774. " \n",
  775. " # 卷积层 '1'表示输入图片为单通道, '6'表示输出通道数,'5'表示卷积核为5*5\n",
  776. " self.conv1 = nn.Conv2d(1, 6, 5) \n",
  777. " # 卷积层\n",
  778. " self.conv2 = nn.Conv2d(6, 16, 5) \n",
  779. " # 仿射层/全连接层,y = Wx + b\n",
  780. " self.fc1 = nn.Linear(16*5*5, 120) \n",
  781. " self.fc2 = nn.Linear(120, 84)\n",
  782. " self.fc3 = nn.Linear(84, 10)\n",
  783. "\n",
  784. " def forward(self, x): \n",
  785. " # 卷积 -> 激活 -> 池化 \n",
  786. " x = F.max_pool2d(F.relu(self.conv1(x)), (2, 2))\n",
  787. " x = F.max_pool2d(F.relu(self.conv2(x)), 2) \n",
  788. " # reshape,‘-1’表示自适应\n",
  789. " x = x.view(x.size()[0], -1) \n",
  790. " x = F.relu(self.fc1(x))\n",
  791. " x = F.relu(self.fc2(x))\n",
  792. " x = self.fc3(x) \n",
  793. " return x\n",
  794. "\n",
  795. "net = Net()\n",
  796. "print(net)"
  797. ]
  798. },
  799. {
  800. "cell_type": "markdown",
  801. "metadata": {},
  802. "source": [
  803. "只要在nn.Module的子类中定义了forward函数,backward函数就会自动被实现(利用`Autograd`)。在`forward` 函数中可使用任何Variable支持的函数,还可以使用if、for循环、print、log等Python语法,写法和标准的Python写法一致。\n",
  804. "\n",
  805. "网络的可学习参数通过`net.parameters()`返回,`net.named_parameters`可同时返回可学习的参数及名称。"
  806. ]
  807. },
  808. {
  809. "cell_type": "code",
  810. "execution_count": 27,
  811. "metadata": {},
  812. "outputs": [
  813. {
  814. "name": "stdout",
  815. "output_type": "stream",
  816. "text": [
  817. "10\n"
  818. ]
  819. }
  820. ],
  821. "source": [
  822. "params = list(net.parameters())\n",
  823. "print(len(params))"
  824. ]
  825. },
  826. {
  827. "cell_type": "code",
  828. "execution_count": 28,
  829. "metadata": {},
  830. "outputs": [
  831. {
  832. "name": "stdout",
  833. "output_type": "stream",
  834. "text": [
  835. "conv1.weight : torch.Size([6, 1, 5, 5])\n",
  836. "conv1.bias : torch.Size([6])\n",
  837. "conv2.weight : torch.Size([16, 6, 5, 5])\n",
  838. "conv2.bias : torch.Size([16])\n",
  839. "fc1.weight : torch.Size([120, 400])\n",
  840. "fc1.bias : torch.Size([120])\n",
  841. "fc2.weight : torch.Size([84, 120])\n",
  842. "fc2.bias : torch.Size([84])\n",
  843. "fc3.weight : torch.Size([10, 84])\n",
  844. "fc3.bias : torch.Size([10])\n"
  845. ]
  846. }
  847. ],
  848. "source": [
  849. "for name,parameters in net.named_parameters():\n",
  850. " print(name,':',parameters.size())"
  851. ]
  852. },
  853. {
  854. "cell_type": "markdown",
  855. "metadata": {},
  856. "source": [
  857. "forward函数的输入和输出都是Variable,只有Variable才具有自动求导功能,而Tensor是没有的,所以在输入时,需把Tensor封装成Variable。"
  858. ]
  859. },
  860. {
  861. "cell_type": "code",
  862. "execution_count": 29,
  863. "metadata": {
  864. "scrolled": true
  865. },
  866. "outputs": [
  867. {
  868. "data": {
  869. "text/plain": [
  870. "torch.Size([1, 10])"
  871. ]
  872. },
  873. "execution_count": 29,
  874. "metadata": {},
  875. "output_type": "execute_result"
  876. }
  877. ],
  878. "source": [
  879. "input = Variable(t.randn(1, 1, 32, 32))\n",
  880. "out = net(input)\n",
  881. "out.size()"
  882. ]
  883. },
  884. {
  885. "cell_type": "code",
  886. "execution_count": 30,
  887. "metadata": {},
  888. "outputs": [],
  889. "source": [
  890. "net.zero_grad() # 所有参数的梯度清零\n",
  891. "out.backward(Variable(t.ones(1,10))) # 反向传播"
  892. ]
  893. },
  894. {
  895. "cell_type": "markdown",
  896. "metadata": {},
  897. "source": [
  898. "需要注意的是,torch.nn只支持mini-batches,不支持一次只输入一个样本,即一次必须是一个batch。但如果只想输入一个样本,则用 `input.unsqueeze(0)`将batch_size设为1。例如 `nn.Conv2d` 输入必须是4维的,形如$nSamples \\times nChannels \\times Height \\times Width$。可将nSample设为1,即$1 \\times nChannels \\times Height \\times Width$。"
  899. ]
  900. },
  901. {
  902. "cell_type": "markdown",
  903. "metadata": {},
  904. "source": [
  905. "#### 损失函数\n",
  906. "\n",
  907. "nn实现了神经网络中大多数的损失函数,例如nn.MSELoss用来计算均方误差,nn.CrossEntropyLoss用来计算交叉熵损失。"
  908. ]
  909. },
  910. {
  911. "cell_type": "code",
  912. "execution_count": 31,
  913. "metadata": {
  914. "scrolled": true
  915. },
  916. "outputs": [
  917. {
  918. "data": {
  919. "text/plain": [
  920. "Variable containing:\n",
  921. " 28.5536\n",
  922. "[torch.FloatTensor of size 1]"
  923. ]
  924. },
  925. "execution_count": 31,
  926. "metadata": {},
  927. "output_type": "execute_result"
  928. }
  929. ],
  930. "source": [
  931. "output = net(input)\n",
  932. "target = Variable(t.arange(0,10)) \n",
  933. "criterion = nn.MSELoss()\n",
  934. "loss = criterion(output, target)\n",
  935. "loss"
  936. ]
  937. },
  938. {
  939. "cell_type": "markdown",
  940. "metadata": {},
  941. "source": [
  942. "如果对loss进行反向传播溯源(使用`gradfn`属性),可看到它的计算图如下:\n",
  943. "\n",
  944. "```\n",
  945. "input -> conv2d -> relu -> maxpool2d -> conv2d -> relu -> maxpool2d \n",
  946. " -> view -> linear -> relu -> linear -> relu -> linear \n",
  947. " -> MSELoss\n",
  948. " -> loss\n",
  949. "```\n",
  950. "\n",
  951. "当调用`loss.backward()`时,该图会动态生成并自动微分,也即会自动计算图中参数(Parameter)的导数。"
  952. ]
  953. },
  954. {
  955. "cell_type": "code",
  956. "execution_count": 32,
  957. "metadata": {},
  958. "outputs": [
  959. {
  960. "name": "stdout",
  961. "output_type": "stream",
  962. "text": [
  963. "反向传播之前 conv1.bias的梯度\n",
  964. "Variable containing:\n",
  965. " 0\n",
  966. " 0\n",
  967. " 0\n",
  968. " 0\n",
  969. " 0\n",
  970. " 0\n",
  971. "[torch.FloatTensor of size 6]\n",
  972. "\n",
  973. "反向传播之后 conv1.bias的梯度\n",
  974. "Variable containing:\n",
  975. "1.00000e-02 *\n",
  976. " -4.2109\n",
  977. " -2.7638\n",
  978. " -5.8431\n",
  979. " 1.3761\n",
  980. " -2.4141\n",
  981. " -1.2015\n",
  982. "[torch.FloatTensor of size 6]\n",
  983. "\n"
  984. ]
  985. }
  986. ],
  987. "source": [
  988. "# 运行.backward,观察调用之前和调用之后的grad\n",
  989. "net.zero_grad() # 把net中所有可学习参数的梯度清零\n",
  990. "print('反向传播之前 conv1.bias的梯度')\n",
  991. "print(net.conv1.bias.grad)\n",
  992. "loss.backward()\n",
  993. "print('反向传播之后 conv1.bias的梯度')\n",
  994. "print(net.conv1.bias.grad)"
  995. ]
  996. },
  997. {
  998. "cell_type": "markdown",
  999. "metadata": {},
  1000. "source": [
  1001. "#### 优化器"
  1002. ]
  1003. },
  1004. {
  1005. "cell_type": "markdown",
  1006. "metadata": {},
  1007. "source": [
  1008. "在反向传播计算完所有参数的梯度后,还需要使用优化方法来更新网络的权重和参数,例如随机梯度下降法(SGD)的更新策略如下:\n",
  1009. "```\n",
  1010. "weight = weight - learning_rate * gradient\n",
  1011. "```\n",
  1012. "\n",
  1013. "手动实现如下:\n",
  1014. "\n",
  1015. "```python\n",
  1016. "learning_rate = 0.01\n",
  1017. "for f in net.parameters():\n",
  1018. " f.data.sub_(f.grad.data * learning_rate)# inplace 减法\n",
  1019. "```\n",
  1020. "\n",
  1021. "`torch.optim`中实现了深度学习中绝大多数的优化方法,例如RMSProp、Adam、SGD等,更便于使用,因此大多数时候并不需要手动写上述代码。"
  1022. ]
  1023. },
  1024. {
  1025. "cell_type": "code",
  1026. "execution_count": 33,
  1027. "metadata": {},
  1028. "outputs": [],
  1029. "source": [
  1030. "import torch.optim as optim\n",
  1031. "#新建一个优化器,指定要调整的参数和学习率\n",
  1032. "optimizer = optim.SGD(net.parameters(), lr = 0.01)\n",
  1033. "\n",
  1034. "# 在训练过程中\n",
  1035. "# 先梯度清零(与net.zero_grad()效果一样)\n",
  1036. "optimizer.zero_grad() \n",
  1037. "\n",
  1038. "# 计算损失\n",
  1039. "output = net(input)\n",
  1040. "loss = criterion(output, target)\n",
  1041. "\n",
  1042. "#反向传播\n",
  1043. "loss.backward()\n",
  1044. "\n",
  1045. "#更新参数\n",
  1046. "optimizer.step()"
  1047. ]
  1048. },
  1049. {
  1050. "cell_type": "markdown",
  1051. "metadata": {},
  1052. "source": [
  1053. "\n",
  1054. "\n",
  1055. "#### 数据加载与预处理\n",
  1056. "\n",
  1057. "在深度学习中数据加载及预处理是非常复杂繁琐的,但PyTorch提供了一些可极大简化和加快数据处理流程的工具。同时,对于常用的数据集,PyTorch也提供了封装好的接口供用户快速调用,这些数据集主要保存在torchvison中。\n",
  1058. "\n",
  1059. "`torchvision`实现了常用的图像数据加载功能,例如Imagenet、CIFAR10、MNIST等,以及常用的数据转换操作,这极大地方便了数据加载,并且代码具有可重用性。\n",
  1060. "\n",
  1061. "\n",
  1062. "### 小试牛刀:CIFAR-10分类\n",
  1063. "\n",
  1064. "下面我们来尝试实现对CIFAR-10数据集的分类,步骤如下: \n",
  1065. "\n",
  1066. "1. 使用torchvision加载并预处理CIFAR-10数据集\n",
  1067. "2. 定义网络\n",
  1068. "3. 定义损失函数和优化器\n",
  1069. "4. 训练网络并更新网络参数\n",
  1070. "5. 测试网络\n",
  1071. "\n",
  1072. "#### CIFAR-10数据加载及预处理\n",
  1073. "\n",
  1074. "CIFAR-10[^3]是一个常用的彩色图片数据集,它有10个类别: 'airplane', 'automobile', 'bird', 'cat', 'deer', 'dog', 'frog', 'horse', 'ship', 'truck'。每张图片都是$3\\times32\\times32$,也即3-通道彩色图片,分辨率为$32\\times32$。\n",
  1075. "\n",
  1076. "[^3]: http://www.cs.toronto.edu/~kriz/cifar.html"
  1077. ]
  1078. },
  1079. {
  1080. "cell_type": "code",
  1081. "execution_count": 3,
  1082. "metadata": {},
  1083. "outputs": [],
  1084. "source": [
  1085. "import torch as t\n",
  1086. "import torchvision as tv\n",
  1087. "import torchvision.transforms as transforms\n",
  1088. "from torchvision.transforms import ToPILImage\n",
  1089. "show = ToPILImage() # 可以把Tensor转成Image,方便可视化"
  1090. ]
  1091. },
  1092. {
  1093. "cell_type": "code",
  1094. "execution_count": 4,
  1095. "metadata": {},
  1096. "outputs": [
  1097. {
  1098. "name": "stdout",
  1099. "output_type": "stream",
  1100. "text": [
  1101. "Files already downloaded and verified\n",
  1102. "Files already downloaded and verified\n"
  1103. ]
  1104. }
  1105. ],
  1106. "source": [
  1107. "# 第一次运行程序torchvision会自动下载CIFAR-10数据集,\n",
  1108. "# 大约100M,需花费一定的时间,\n",
  1109. "# 如果已经下载有CIFAR-10,可通过root参数指定\n",
  1110. "\n",
  1111. "# 定义对数据的预处理\n",
  1112. "transform = transforms.Compose([\n",
  1113. " transforms.ToTensor(), # 转为Tensor\n",
  1114. " transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)), # 归一化\n",
  1115. " ])\n",
  1116. "\n",
  1117. "# 训练集\n",
  1118. "trainset = tv.datasets.CIFAR10(\n",
  1119. " root='../data/', \n",
  1120. " train=True, \n",
  1121. " download=True,\n",
  1122. " transform=transform)\n",
  1123. "\n",
  1124. "trainloader = t.utils.data.DataLoader(\n",
  1125. " trainset, \n",
  1126. " batch_size=4,\n",
  1127. " shuffle=True, \n",
  1128. " num_workers=2)\n",
  1129. "\n",
  1130. "# 测试集\n",
  1131. "testset = tv.datasets.CIFAR10(\n",
  1132. " '../data/',\n",
  1133. " train=False, \n",
  1134. " download=True, \n",
  1135. " transform=transform)\n",
  1136. "\n",
  1137. "testloader = t.utils.data.DataLoader(\n",
  1138. " testset,\n",
  1139. " batch_size=4, \n",
  1140. " shuffle=False,\n",
  1141. " num_workers=2)\n",
  1142. "\n",
  1143. "classes = ('plane', 'car', 'bird', 'cat',\n",
  1144. " 'deer', 'dog', 'frog', 'horse', 'ship', 'truck')"
  1145. ]
  1146. },
  1147. {
  1148. "cell_type": "markdown",
  1149. "metadata": {},
  1150. "source": [
  1151. "Dataset对象是一个数据集,可以按下标访问,返回形如(data, label)的数据。"
  1152. ]
  1153. },
  1154. {
  1155. "cell_type": "code",
  1156. "execution_count": 5,
  1157. "metadata": {},
  1158. "outputs": [
  1159. {
  1160. "name": "stdout",
  1161. "output_type": "stream",
  1162. "text": [
  1163. "ship\n"
  1164. ]
  1165. },
  1166. {
  1167. "data": {
  1168. "image/png": "iVBORw0KGgoAAAANSUhEUgAAAGQAAABkCAIAAAD/gAIDAAALVElEQVR4nO1cW3MVxxGe2d1zk46EhEASlpCEBaEo43K5UqmUKz8jpCo/MQ/Jj0j5JQllLGIw2NxsK4iLERJH17PXPEz316OZxdLoeb4XWruzvbPDfNOX6Tn64cuRUkopVde1OomqEbmsaqcZhIKbFTVJVVV5jRsWRGdRlaRc8d2Gbmtu3/ADTdM4Ql4m0tXavYs+NI1mVepjX9pUckUXlXMX7RMVcWbEwQpAppkCEACttMgsJiw1uOK1EYHvJWhtvQWqUhY0s0FrJqbGYy5V00S6Bwjf5RpdSZKU/vaorRrpldau2oRfFGdWAOJgBSBLZMLy5OS/Ey1DCakRXqAZBDZJunEayRVrkguNEpe3iSwO3DkxWCCv9R0ed1J0hvsO9uFtYLTSNg3d5QhsjTMrAHGwApBZfHJnvj2zwYtaYTLz5GzQ5qQiS2w8U6tsHmm3WeL1wmK9e+VEJ+C7ijlkN5VvZQluJM5HqTbvF0Y6zqwAxMEKQBysAGRwWH0XO7GMqCxVWNr4AawFvivQ1O6CaLvdWpNLXXNEnYgn7boC0r0Ga6ulSvrAjgVPg6pkj58vQUOt8S2W68APIhiIHvx5EAcrAJk18wkWc+ygF3EsLnLuSey9qwNudC0+hJU5Ep/Ddf0tEvqeDFwHiztYHBpXgzg0csv3VERV43EzevDnQRysAEg+S8JgvldbbEgSmBKka+mWzHyhFWJyNIJOC4hs4VJLPotttJDPzZbZy0cj7V2zqFJXuZh7lmp7zqAPSMYpV4g4HXGwAiCBdIt32tjWENlk7d/FE87f2mOHnSxqGtCQA1rtPqi8KxBOmjD0wPOQvTy45fm2JA1ASY3eeHyMOB1xsAKQiY/n08q68BteX4vQss9DQm3lfxEb4lErGhVb62mSXLXVP5ek1h0v2e0pP/HtkuJ2I9w4swIQBysAWcPTrvIqAM6I1CcdiMIzuYBHmGR4MEFWl1mQ8pNlUzhv0QolCOzxCotV3fD/OvIwKLNgVbVmd9oLRWtRLouD1q6vHGdWAOJgBUBI0VKMcDYknt8olUaNn8axd114ejeunRKC1O5mkrWDeyI6dL7DN82+oYQ1TKz8rV8A4Wd7Ik5HHKwAZFq5+/26JegTaEmQIg/jjnjLhEey0dr8hJXBbgjebO3XukkebM3aqkBkLQkc3PXKJrxMqd3hunYXkxQ+s/tVER9HHKwAZLLm+5nEVrRUyzGJavdOW+7UripgS4dSPNGIJCZtKaZI47CG1OonzC7KpPx9B+isGrwOHRUaViBpTZ5qmqb8FRFnRhysAMTBCoCUSYK24H9b4liWNvHFUZXfkrjlhYP5n1mrQ8aBcCUmPOFuUbMclf68q4QOpFYVJ+JorINwgLAdlXiOQtW6THsZ7RhInwdxsAJgbbJaRbzm38r2tgE/Nhan2fWMMZNB0IP9D9C0vf3OCEXB2StW1ZuYcl47nBxSr/hQTpL15TO482VJroZfoCDuiFecUdvBAF/WnBqLZ3fOgzhYAcjatm1cwYb2JrNVe8R/s+1Dk4QLfp/98BCq7t69a4TxeGyEPCc+Fg1Zyi++/NIIn9++bQTQcHK2B1U4QqekNApW3k1tV6UbFdi5A1hPmGZvszXiDIiDFYAsscp86N/W3BPguay1xmRmTR5/Gy6xXbh0ERdXlz+hFzEdtt+/N0JeEw0zVvr4+wdGuH79Bt868Qb+CPSKbTrTFoF3guJcvlLZpcbMOkmXt5Q2RpyGOFgBaNndCd/fYQ1ylo6Jyf8X+TGZuV5X3njzxroRpqbIBf3mm3tG6A5njXBwdER9YtZfnL3g99M6g4cKRWTZvEIoT0r8PLhStXcsPs6sAMTBCoCYlMqLqqQw1vb6pJCBnTdVOQ+CAjjR8fbtKyN8d/9b6Dw+PjbC5i+/GCHNiKTXrpOw9XLLCF999SfuFPWqKqQeIvUOi9f8OR22ffiZCvldB8mQW7UOqPzDOHBqO86sAMTBCkBWeT+XIlV6lt2QX3GQ/U9qX1aF00YOjLEvOneZrJvqiDVMFQV3U3Nz1GyOXNa8yo2w9YpoOL+wyMq5JMi22rUwivopd9wtnFq5YeOJPSfv5EyTRGsYjjhYAcgQOllzklDVYiPQLFMwgkix8hlLsaL0f3BhetoIPzx5YoT5K8vQeXBwYISpGaLh/v6+EV5vEfuevPjJCH/7+z+M8Jc7fzVCryuZUuvnlOhKXoBE2hFg2cUVtew+fNESzWKtwzkQBysAcbACkB0XpXNJ9kUsM4/cccXubJmT/52mXW5BQ//zTz8b4e3bX42wf3hohPxEJRScD96w6Q2MsLh01QhXr103wmBIy193YpJ7YvWZ/Ymyoe6N+St6aYe/y1udJeQQVVhwk9oNSOLMCkAcrABk9+7/10jwtuEldKzcU6/DfnNN/vrkgPzvJCEaNglduXdvwwgbG/eNsLu3Z4SF1TXoXF4mN+Lp06dGmGNXfmVlxQjrN24aYW2Nkl9vft02wrgQHoJZ45w2ipBTyziQxg6TtfdLRCtKey1q4SZpcC9EfBxxsAKQvf+wa6TBgCxRxkmlzLKGmoPJNSbIzDTlgvsDqkJ49uJ/dGuGMr/r69eMsDMi13x6fhE6//Xv/xhhc3PTCCWnqO7c+bMRZmcptH786LER3rwmGua2OWQTdshmt9MhIwinPpX9Hg6k4dNbNMTeKtYlv4Y64nTEwQpABpNSHNAEnp2l3FOv30W7hUt0scPcHI12jbC3T/Gw4jNqv7tJlmtpiUi3u0c03DnMofOPf/i9Eb74/DNqtks6+/zqmRnyRY8OaJvnYH/EfWeiWdVRiIgrzohhdwe0bbyAv2yj4W9UL0WcjjhYAcgSnszb22Rl9njCPzvaQbseVwpcmiVepFLaQCPe53I9mNGq5NxQ2bJBsrJ8hVRxVT4MMRzjfEz28ZPFy0bY3KRUV29yILqYUKMRkTTPmYZcnIsMV8qVvzCCRdFCQ+tcbsxnhSMOVgCyhmfdxUs0z1EOW42lWLbhY9mDASVzUQePCp5KUZuDQ7KPBVfyjXMOPGsxYTnzGDSE3cmYKSknWLocga6vXnUeV0qV7HlWnDhqeM8JDNOpe1K8kjNDkjgqeenAmlDHFM05EAcrABkog1mHdAccQqWULjkvyns5OVfN9jPKzHSEO8je8OOY+aX1Yww1NjvlPdyM+ctv2d+jDmRMzP60dC/nOG5+boaUF2TT9yoUPXT4HbKBRVcSoXQxphdVXAQMWxlnVgDiYAUgO2YaznEyBDwBv5RSyyuU1ex1aTI/evS9EV5uvTHCYEhbCUh4dlLyG3WXnUxl5yS50LxyDWuGA6mcGtIDEsbwNot9UcQBYMo1VDOTE0Y4PqRDL3VO2VosF3ND3h9ZmIcq1Dq8eU0PVtXgRHcjzoI4WAGIgxWAbOEy0fWIyzQS9iFu3/4M7VaWKTO1NyLmT0xQNvnwmIz00xfPjfDkx2eknVUhRzbJJ+GU5a9P8PrS4aieM2MSig/6tHCguPKoOIYq/KbTaIeC//l5itKHvJIOp+gtV68sGGHpCn17t2M5NLwX++7dB/5k+sA4swIQBysAGfI+MMljrtPf2JDK4offkYBULJJWq2trRrh165YRUGb14AEduHn+nBi6s7MLnb0eu/68EwNh0KFb3Q7Fz91u12lTWbWNSUqdQeHFCgf8K4urRri6St7PBU6E9bFzbKnCNm2vR+m50ZAS7nFmBSAOVgAyJGum+QDN+JBouPVqE+0O93aNAIp1mBf//PprI3Q9WoE7S0tLRsjzH6ETaazhkExkxldqjl1hm0bcAcTkCJ6VUkfHtIZ8yiVKO2wWYaw7XVI+9SkRM0mQ/hYavt+mF/X7ZD3n5siUx5kVgDhYAfg/pQ4eZ65sAxcAAAAASUVORK5CYII=\n",
  1169. "text/plain": [
  1170. "<PIL.Image.Image image mode=RGB size=100x100 at 0x7F1EC53B6588>"
  1171. ]
  1172. },
  1173. "execution_count": 5,
  1174. "metadata": {},
  1175. "output_type": "execute_result"
  1176. }
  1177. ],
  1178. "source": [
  1179. "(data, label) = trainset[100]\n",
  1180. "print(classes[label])\n",
  1181. "\n",
  1182. "# (data + 1) / 2是为了还原被归一化的数据\n",
  1183. "show((data + 1) / 2).resize((100, 100))"
  1184. ]
  1185. },
  1186. {
  1187. "cell_type": "markdown",
  1188. "metadata": {},
  1189. "source": [
  1190. "Dataloader是一个可迭代的对象,它将dataset返回的每一条数据拼接成一个batch,并提供多线程加速优化和数据打乱等操作。当程序对dataset的所有数据遍历完一遍之后,相应的对Dataloader也完成了一次迭代。"
  1191. ]
  1192. },
  1193. {
  1194. "cell_type": "code",
  1195. "execution_count": 6,
  1196. "metadata": {},
  1197. "outputs": [
  1198. {
  1199. "name": "stdout",
  1200. "output_type": "stream",
  1201. "text": [
  1202. " cat deer horse plane\n"
  1203. ]
  1204. },
  1205. {
  1206. "data": {
  1207. "image/png": "iVBORw0KGgoAAAANSUhEUgAAAZAAAABkCAIAAAAnqfEgAAA09UlEQVR4nO19SZAk53Xel0tl1l7V1dvs0wBmAAx2AiAEimCQIilrs6wIy3JYctjhi0+KUIRPvvrkcPhqRVg32WFH2AeFFZYoOyTKpCRCJkWCBEgAxDL7TE/3dE8vtW+5+fC/72VNVTWla9P/u/Trqsw///wzK/N9b/keYMWKFStWrFixYsWKFStWrFixYsWKFStWrFixYsWKlf+/xFn86Lf/2T+W75zMKNFkDGDc78i/45FR0mQq+6Ty13VdjitKPI2NksHnMV0eKpvdOQgLRikWQ6P4BdnF90UJAtmmVCoZpcCvCp4MW3AdAFkmw2ZpwoOlPL5sWQzkQOVi0SiVUpHnLisTp7II//zf/AfMyMVXNo1SXz1rlO7BUHaJZBcvkANVm3KgUi2QyfgyfhRnAApcyEd7B0bZuHDeKGkk3/WPxrJv7HK2Mlp/0DOK68n61Os1AFEki58kMScuc5tOp3Nn6vuebOLJJ54vB/ID+aRYCgCMB3KmDu+feCrDfvD2D/G4/M6//FdG+fJrbxmlfSyzvX94DOCv3v66+ff2w4dGefPNrxjlaedYxu+Jcu7zv26Ug0jO/c7tD41yeHDDKN3jXQD1alXWpCpX6v1P7hilc/xAvsoio3zx818wyo3rd41y4dIlo2zfet8ov/qLbwHYurhl/r11Z1+mtLpqlJsc/79+7S+M8upXvmyU15590ijR9n2j/Nvf//eYkd/+F79jFL3VPU8uhxvImTr8KsuyOSXgnex4sk2SJLODeK4oSSw/h6AgP4d4MpEDpforczm+bDOZyP1jfkxONuHEZQM3rMmBQln2zJNpT/UmTKc8Ndk5iuSWTibR7OnEsVyX3/29f4fHxYUVK1asnBLxl3wWy9ssLPKJjgEAL+ubfx1XHpkFvpML+oB39QEvg43G8liNJzRweJwgDDBjb4Uh7Z1yyA3EXnBdtQJ8blyYG81xOL6TAfByO27+lRUEYkYVaa8FtCmcTGbrOnw1LTNCAaRTOdz+jthEDt/5NFmQuLJN4vBt78uhx7EYKYVKCKBcF4OxnkRcBNkygyhRJGfUO5R9hzR11WZUu2k4HIGvWQApzUzP05dnxtk6/ETPja9NDuupFea4AMo0b8OCLGC708cJctyVrz65c8soMV+5e8dHAFpnG/J5IMe9+f4PjHKxJed+jldzjdbZ+lf/vlE2X31OTqSzY5S/+J9/AKDIpdgdyXJ1Y4EIo1he7Jc3zxjllRc+I1O6LVbed97+c1kMyBX5zrsfA3ju+c/KuTuPjPLBt75plMBvGWV15aJR9g9l38++9kWjTNY/kXX5fcxKgSupii6+S5zh+PP3fERLJDeTaUl5jgcgpYnk8YeZ8cZOE7GS9GrGEUdz5VYo0GD3qGSpAwCp/ELVcnf5401dudNcXw5d8NXUAkeT2QZ8gMTeFECaqkF3oiFlLSwrVqycGrEPLCtWrJwaWQIJ6011bItSqzoAJiX67cZi29H6QyEoGyXjE3A8IRih0z1OxnMHCn0fQKFApBaKoUhXMqpVOt2J5tT0dR0ZP8l96iKu5wFwGTFIErEzpxOZ9ngkSMqt1eXQBKGuR2OYs/LS5c90n+7GCaFHRsjmevKVz+jBeBRxMrKxH8qy+BUPgEuvdq0l4DGhP3hIS9ohCi5VK0bpHQsqqVYqevbmT7/fBxDRyC8QVqiikDCHHlSCgLibG8eED+PhCECrIeumoY8pgf+iXL99xyiNspzaG6+/bpRPv3ULwL37AhVrZTmLF56/YpTi4Z5RDnZ3jbLaEVd3CXLE5oogyj//+h8ZZWf7AYDLq/L5rR+/Z5RWTQDmL/7SV43y5OUto9y9e90od7YFsg0GgvTf+rmfN8pzz70MoN2RmycZDoxybiKfxEdyOYpDXjLePA8PZNqF6QDLRH0deofnCiGg3oger1QKdWWAG4sYL0rG+E+qPhqNhjEM4/De0/s2pmNef3cakJlOYgCOw9splN9qyhlkKc89476hukqIRhnIcghgUSgASDgnxcWLYi0sK1asnBqxDywrVqycGlkCCTvdtlEGPTERy2EAoMCYWsEXIKAJQUFBsE9EY9IlUitWxCB0mdyhIa1yuQSgWhM4WQg0nMc8KaZHKb5ziPw0bqUBRM9T1FMAgFRtVDmLlHPTUJpLWDEYCFyNmPRUDJmZlaOtx+TlF58wykFHhr13QxBBjZh2TMw8ZKCqALGB11uSt1IsVwAEFYYCaQyPOKXpVAaJafa7BVmfUlmWztXV4O4m4BLHmoelIVReIIIHDc2oOD9ByYCZ6HBCyKlJT4tS1qw6Xt82b7B+rwPgledeMP+eXZEo29OXn5adCbI0D2sqgyG+I+lXLhfh7gcSW5xM2gDe/sF75t8Wces/+YVfNcoLr71mlGuvvmKUvY6M75bkjP7HH/yhTOaZa0b5/OfeAuARAP7NLUn72rwuKPJi7YJR1mJZsG4iMc3f+0+/a5RnN+tYKry4izlWWaIXiBdR8+MYPNWL6Co2dBzMYDoN52kcX38vE8L5IJCNM41up3Rc8CfpSf4gcx75w9cDezz3mEeMuGKe5pEl+tvk3o4DIE71Ll0emoe1sKxYsXKKxD6wrFixcmpkCSTUGhePj7MszgBMIwkVqcHm+VpownBSLNtMUwFBil+qRWYbEm2Vy2UAtTqT+mm+akGAT6tVUxlT2pmaCanpjppKWqvXMRNJUYs6Yb6ixs6iCZNa+ZXm14UEMmFYwjJ57bWrRrmz3ZUtCfeePHfOKPcZG7qx25ZTpjF/9axsY1Lw2jyv9khAhKafJoTDKVGcmvdTAtgC1zbT+ItjJi9nMWH5xXSqRRXgtEVKCsA19sSvKiWBXWaRc5RNyOA4J4Z11taaRtnbk8TOo0MBehurGwDOr0tpy4U1Ua5eFUjoeM8aZTiVZfnff/onRgnee9soFy8JNj/bkEW4cPYqgKeeF4B2sSwobIPKB++8Z5TWplTtXHh2yyi/9Gu/bJRvf//HRrlzS4p1PnttD0CLAeWtp+S4hSPJNT18IOfVLAu2vX5fynp2+nIn7N1hCPxx0QxPF/orYBhXf27O/HXRH4i6YnLJzC5yv2nQOckYIs9kJhEjiQoJC76WyLCujuE842+JCeXiTBMGOKlMA5e8b1mRA24TEkhquHASjwGkC3B4UayFZcWKlVMjSyysp69KFsx0SL/vcALkZo4aL+omd7W0kqk0SWNFNuZLQA0rL3cEZgCCAr3yno7G8TWDA+qApBeZVpLPhCn16ZpH/JiedbXBtAp6PJHzcjL17osFUVY39kJ20pyUCzLa5qoct/WqrJvHhKyMGS4x1ydgOkyDFa2mlGHKtQ19pm5FNDNpu3hMfklY5JSFjDPQwp3wkrlIAUz53o743vNSrRti+hg96C6vS5V5XuD7s8BainIpBICMeTq+XtMTLazzGy1OUk7t1vU7MoVRAqDsyOdFKh6jOgnN2wlfxccDyWOaHom91uu1jaLBnEmcAXj9rc+bf+OurMlfv3fTKJvnxfjqj2Tdxn1ZqJXqulG+8BUp/dn96AM50MN7AGqbcjqdqWCIqCEJX/sPxIzqemLFNBixiVzZ6+jhPpZJHjtayMNSh7pDsJJnI/KUk3T+JpfymnxLOVCs151GjAZdtHSmzPvK02xHpTlABoCAJI8IpTS6tKbNDzTHivZaHhAghuNt4yYuZlHRQiBIxVpYVqxYOTViH1hWrFg5NbIEEl48R4Kno7ZRBt4AAOjXU8tNszOU8ibjeG6gSf2CfRIsFNNkGYBuR/ypmtNRqQouU5+uFotosYvrBJyDAkwxI0fjCYBBn9wSPK+AhSYBvcsFzUnJa9mVBouwa9GdaY7Ls2hWiETo9e/t0xPJ9XninACNpkcfN1FVr9OfPS6JMBCPGAcgmovpL89YKKPrr1k8RZ7j+uoKgDt7h+bfPmuDWEKPDVa0ZDnTgxJFcJGhaJHu0sDHDLWZOk0znOglnXZlDm984UtG0WS07bu7AA6PJAfq/v1t2aAo3vGXPvumTJuXu5pwShW5SWLClg8/IRFCWAZQKEswZ0jYcv+RHKgliA3PvS4reQUyWi0UV0ahQIINrn8hiwFUSrI4RyQF+6Pvfs8o62QBubb1lFE+/8JLRvnu9z82SufBMZaJGg6aH6cgUf3xM84N3pz5AKw/Y2TMoEV1v+TYkIM5iTpGZJsolhs4JvFJyF/xJGJBUpICcDOlOZGJRwpF9TGhhHTMuko53+FYf9d0KTguAF/DSpmFhFasWDn9Yh9YVqxYOTWyBBJutsQqrjOLZ1gdAQiVpY9W64Qg5ZjgUaMGoNmvgYAiI0qaQjUZTQAMmBg0HIjF2Okz4csXJSxLtKhcEaVGIgeHkE0Vkz526axQDIdFmYCvBQ2F+eBjzuhAbDshPoqTEyAhUSSYgtTtMhoyFmXcE0N6tSX462yzKQciJOz2RwAO9yWXxyNcbTE6eNBlNGogkZppVyJlIZc0cpQxQuz8V178WQDXrslR/uzP/8ooTcLtr35JCIvffV9yhXZ5EZXRQa94HMv43V4XyyK/IxJgLEp1Iti8vyO1LI/274lyuA9gvSHpVy1Glm/cFwrju4/+l1HWGGlqEk0cfyQAcJ+h2CtPXjbKOC0AuPup5E997itfMsozz79olHu3BHse9+WUD1mak2Yy/kZVbpLzrwmsazRCAB3SkHz5F/+BUb7+7W8bpcKg9sULQuC32VozyirvWxKgoD2XD6d8ivzAXQjwAcpznW9l/ii7tSZkmWRGdb8U+FtWMmWP/paE0buUSFAT/RLNBeMVTxOjsMCLBA8e102nnSmDZjYf+5shj5RNTA7gdKKELjYPy4oVK6df7APLihUrp0aWQMLGStMoTSomUKI1LlofEzE5U/vcDAiCtM1GwoyyiEUnEakLsmkEIIsFpAxisai7EzFwh33BPqORlL8orNvckCDOuTVJydtsSWhppekBWFlhNUlRwamS/ykdIPgVo5BEi2pa/62QUDnX2QcHGZNaPe66UiVToJZK5PTVCYAC3xxPnBWWcZcTOFORfQ/asm43yCE3JnauNORkJ7TM723fA/Brv/Qr5t/L6wxTrgjsOmoLCDo8FKa6ep1xQ8bF+gMCPYIFzw8xA6Vz3osT0msBDNvCpH58IJUrF87IZFbX1wEM23Jj/Pjjj4xy8drLRgm5Stu7km/5Aq/y+gtStTM+FG6/ClM6G/VNAPfvyS4txhM3tqSY5skLW0aJSSTwta8J+V/GOpJ2h0x7vPf+8v0fALjy7PPm31/49d80ys+Qj/DqmQ2ZCQQA3iSjQ28oi1Cpyt2Cx0nwc6597UDlqAdDy2Lm+fg11qxMeCVl1EvS2S2VfV+pH/w8XEhXA+vqYtJYxmQiVEY9zykASBhhVEioPyUFgooN9YhT1oeVeEs3ecsdHR4AmE5l2FTZBRfEWlhWrFg5NWIfWFasWDk1sgQSBsRHga9+/hTAhFTuWn0U0Fht1gSghUwTHXPjEWHFhJGFlEXbncEIwEFfzL9+KltGzHabJGTL5r6ahloo+3OKdspaWasACAKdpMu5Efcpk3Ruxyoduxask8RaK6weF4e4KSyIkR/4smWzLhVkBbLgN0ge7zCaNuqLsr7SANBsyuQ3G7KSJVrUXeZ8VmmW7+9ISHFCtgYFmCBQ/fCj6wCe2RKQ9dqLAmQ0IfYP//hrMj7DtAGp5fpMiewTxWsqabNRx0x3Jq1cUwKMRVk7tyXbMEHx6JEwtT/zwmcArL8mKHhIlNLcbBrl6lXBfUe3BVupYwEs+mutCuLbPCMVgoNHfQCIBNPppdRwVUFf05z//o7EDT/4kbAA3r4tHU/diYyzUgkxUzR35ZrEHL/0pS8Z5RxzcUMSW97/xjeMsncol2xtQ6Z94yHTXAHMdNZKcyYMzlEDfwTmmqY7Q/KnsTm9tzOzhflvRAZ6OJojrSx9DAWy2DN25iczJZ1k4DqYCTVqxF+5GWLmCSRsWJdxtJjg8Ec/lG67AaPMzVoNQIMI0bMEflasWPkpkCUWVveR1FIUlbPYzTDrDsyrrlnfz66f2vx9HMrzNSQtwTGrUvYGsteN4zGAg2N59mfkBigVWQwB2bdFTiUlzyoVlZ2ZBlQow5arRQAVdoEN6EHkqycnCdLnOK2l3EGoTlD3BKanUN3zXJZqhb587YBCn3VA0uRqIAZUzFfF8xdXATw6Eg9xoFSztOw89nbd3tnj3MQmrbXElJuQv0EpKiJkAO7tyS4/87q4sTsHbaPceyi+9pjXZcAUJ59xhipjBaOxuIh39/cAnKMLX2uDFnmWVe48POQpy+Sikfj7G80VAFvPihn11pelmU2Bfc+P9o+Mcv26dNaZHstoEdOIGrRZBgOxkh492AOQ0L1drMuaH+2Ie/5od4erIZ/84Pv/1yiffCo2qVaYFJj9FDoJgJ0dyRF793t/Y5Q3f+ErcoJchAJtijL79PTpud9Y28Qy0Uw3ZTLIHetsvaNtcd0FA0T/z+iGN/VzZabdqVc+oZmpju2Ye2esRtKKnIIrN2FF+c3jDEDGQfo03LS5vNKaJ3ThJ+yjo7/QJxn9UBe9AXPa2iezpTlWrFj5KRD7wLJixcqpkSWQ8OE9Jraw4U29XsEMd4JaoB7tTFf9mjn9mJL8ySCHXUlF2TkQn+4kdgFkNLmVXnk0lEFYz4BaILZinYqWzrgZU0Los0/HAYBxJIOMdNp0EIbMVamw14vOX7vmKEeadricE/XcT2kMZxBlQst2TKu7Q9q51qrglwsMUxS9NoBGRY4yUpIJ+kRbdZntE5fFOd0niPtgW/DRtM9UmqkACdNsdUKs21yT4/aYujVgrxS3JPhrQndph/wZDqvntao+jscAuj0SbHgnOkdV3v1E/OUXL8ocilPBR+P+IQA/kMt8/oKcYIEgaLIr94w/0YagbL1z2DZKEpP3cVUmY9agS6dvfyizPbwpxTo3PpRqpEe7UiT0YPcux+d9lTunZTKDKAZQZMnU994RkobLL0lA41VGNvbv3OTZs6sokf5owKDB4zLT1xZU9KfEHxeXJVuAhOrYVn+N2SLn2mTExuX8yxXxcPdJ0nA8ZksnJmAWMrkbpwPJoZtMepipxNLSH522/pzdPIyQdwQyf6ukyUTOYx4BmHLxNb9sUayFZcWKlVMj9oFlxYqVUyNL8M5wKHZgmpCeAT6APgtl1GwOGUasVNkRh00r4bIhKLP7t9lT5O49ApnYx0w0pErGtXpJAEhYEPNyheOvr0v5Ra0lympd7NiiJwca9DuYyb5xaSS7uR0tn0TEbhrjUv4JNdE15WROYkKqjKFMJVHrs+i8Tf6G47aE5FbKZzltOaNPbj4AkKRy3GaDsa0SQ7S0qMuMh5YZsskmpOXjqa0QxSfRBMDgSC7lkPHKPRbipAPBZR6nXaoKa0KayCC72+IcWGnIpVlZqQPod2XYFjOPPP/EN19Kn0BQk5jmxaKUB7368lUAXk4OIUihvy1RvP2PpHVNlTU0Zd4bet88uCf46+HtT2WbcgXA5nPS1ijal7M4vCmjjXYl5lgjkirk/HOcNjPyXIa9xnEEoEs+jyIjvy5z36rsLXS71+EocoGe3LpklEYoi4wf4zFR3OTMZzYpJMwWOtvqXZryk1SDjCkw08HX4z3pE4WlU1nSfibeiaQoJBNeWcD7MdPTpt0dzsoDWfMB+FzAIG94Qy5PbX7MWyPlJ0oCkUQKhF3M9HBwbWmOFStWfgrEPrCsWLFyamQJJFxdbxqlxMR5x/EBDIdiXo6HgobG2nNpLAZnpaaEYWVurNx4rDmg+Voo+gB6x5IcyARJBJkAkEpLBvGVfqws5muT4bZGWY5Y8hjgy0IAYcCET1r7mimX94MipcSI+G48Fmim/b6Uo25Oxiw9B+uTQmZR+oyd1aoy7SQRADhiPbpWSoxHCYAS8UWZLVFdBpjiiIEYZuudYRDncy+1eEayF0OL2Hv4CMBKUzbY2xM83u4IWllbkaRQXhYMGMYtsW9VifzlKRNSjx4dANjYEFyjeDnDyTZ8UU7tY1IX/PyvCb3BM09fBrDNrNRGQ67pp9els1bnjqRxrvDmCIkmzgSykmNfoM0x+24V2n0Aq4cyyZU+Q43rcl+V+3KCh8RuVfLxt+klGKrfgBjQdKXLpszaXZdbUdNre0cyWkz81Scv4OsvkwWwJIwOf/zNP55ZJBQYdFYSdk3K1XCbo9wM2gEMWqyz0McUKQCf9wxIrK5ga8LSpQ6rnZIVuVsmKUGoNh9jr4EbN+4B+PimZM8+T9LEl648weMwYJ0uJKbm1WyEfgzBGzBYrzNRmY+aRbEWlhUrVk6NLGtVTxdusajPeAeAS3unHDIzpU/vMsmV2m15WyZa6co2lnsP5f0zINVvIZwAQMYu6rR3jpkMNY3kQTukk2/EYT2+1jbpfW+xbNjHFDPe04h+9DEf2wMahuOpvkW1YnPB05ktf6Z7bDZZJhfYZExCZLLrliraKV4+GfJt36CX+vlnrwHotcUC8hjQ8D368nmJGnWx9Wor8tXZqUwy9cV26LFItRIWATSaq/xc3qJHtLBSfRXz3a6l1HFXNl5pyiIXWREVTX3M0IepBEGAE+RzbwkXc78v5/jaF6UEp9hcBbAK8cEXQ5nShz+WCuSDB2KU1WM5rybtWY0VXGjKsly5JMbL6HAA4AKr4je0UQz9watk3O522kZpVeUED5iMdsyrGUdaWRIAWNuUJf3q3/uyUc6fER7k40dSX6U5ekX1x9PMLCz5wQHIfe1qAemt6LustqHTWkv01dQKNKGJ47lZBlbSYLbih9dOMUSFRWvdqfwutCYPzNV6+1ti6t69swPgeCA/80PS0o0yiSo4tIG8sCljVGTFFPRo/yo1JzuDIwD7exIeGVoLy4oVKz8FYh9YVqxYOTWyLA+LVE36nfH/RYRsKZ1/PtmHayTeTRJ5Ao7HtMO72leDdiyrakajPoACwVFIh3qpqA5vgVRFuqKLbIFTKimbAksB+PA1BmfO0KTVQvTweXQVV7WhjnJaaWcgxZQn1I0XmRDEyvac+6FAtDgZCpCZkHOqEgqAvXRRmvrEgzaAY3p2o5FsqcsC1lJoN1kFAjFd+IMRIxsjjYf0AJw7K2lfIZ3KUawxEIYgCoovmHlERDMecfyuzGpzvQWgypImbWIUs7frovzWP/0tDivw5PKWnLvpxXKmIJf71qfSbfSDG8IVdbQtFTMXuRq1TWE7SOl9L5NOY60mN2FcrAGo1mXYR7vCbPXJR5KoNSTFmMPLvcbOPTu8ZC4LxTKWXpkb7CyJmL/yxZ81isKx7i69/hz2+avXjNLYENh4i3Vvc6K9ncplpra5giJz9iulqUoUPyoReTz3leFWiXir5B1VSUhy3GOOHityEJA3hX6JoChIOSzLWZ895wC4XJGUukuXBAm2zoj3vV6Se0OxZ0TqBeVGPz4SEpFety0nkk0xQ0euNFuLYi0sK1asnBqxDywrVqycGlkCCaeMLEwZp5hGJu5Gur5qg4pYjIk++BjbirnxOqOE/cJ1o7Sj20YZPRoCiMasHqDpWGQNxyYN6SAv1pHxNRDj5HSxGukDZiChdnZR/jNtFAqCIKa8zNQ+6Pje8md6ljebFCVgxUycchhGmlihgc2LckYrqxKA++DuJwAc2s8xSSYU4RbyfBzwE05b6yEYCa2xjmRSLgKoMkx5/rxgww8/YR1GprhSJjemHc7OpFhZkdk6KYno+gMA00gOd+GCgLufUJrzMlOQVBSemHKUkP8ePJJY832SC2rRfodT6uXRWy7CiJlTd6R8pLXaALC+Lud+f1foie9QmTAoHDJPLSSbhcfqkFIsaDfq0wHi+QCeufK0+XeFfL5dcgpOWD80Q6cnylki2fF0OdjJi2wSjQmyIQ3ZLHymBCqDiLLl5UuaQ0IHM6zEE6L7eCJh4hFJLHoD+SR02LSGkLDWkBtg66nnjLK5UgbQWJHlqnEBU57X8EhKvro9iZl2hqLE7PLravkOc7WMOyK/1W0jVStWrPwUiH1gWbFi5dTIEkhYaYiZp3mPpnC835MgQo+FLKknNl6ZXWEKRY2yMWanXWccBpIcEkVXygDqbH8CQsIeCyaG7LhTqgj2PJ9IZ5SnrjA/TUNmkQbIRpjpdqMxNeUwU0M6TeYTR7W5iEOSvDhdbp0mLM1xq5oTyCjeWJQaeeWnicIKmczDPSluMB1KHH6eKjbUYn0m1irPtaN5gwqQtXyHgaRi4AGokGu/wRTQ7QeCm7S0KCI2TPz5dESfk9GoTbFUAqB0jT0y+fmFE5n8ZhgQFbyzVsMMS/Q9ILlgn6nFLinrHvEE7/PemxKDTAl2AtKwnxuPAFRYYHTILMQ9Vswocz+YRhtqDQoD0zXSjYwnMmytXAPwmZdfNf+WQwHX/Uxqy3bJGZ8yAqvhVPVLrDKePicjlrsNSJWn6+b6RIKFcO6rvL8pu0npNsVyDUC5JouwvnnOKDFTcDUMff5YLuLNmwKZXaLR1TXZ68ozguvrwRTAwf4+j0soSq/HjRtCjuizWssrsH+VEmDwvsp46Q0dvkYJs5wrdF6shWXFipVTI/aBZcWKlVMjSyBha0MiGgGLCk0VXrE2nyLps5pfa80URGh4osoenGsrEvtrkMPg3nEXQJDI59WKZPoVmER3/UOpYAqKrBPUNmJklY6mDKuR8KBWbwDItCElZ5Jno/Er7QPqERFUtS0SIaEyEc5JwvaWU6KVKGbCbSZnVCTl3gpP+fBQAmG9EakRyg4AXztWMripHGlFXW1meCYsLnOYSlrOiD0JPQyTX5GZvT96/0dUPpRzpzVeIe1crMHBFiOJiYw2YcrfpQvnMVMC1meAyTmhGRoe61u1PAdXIfeEfIQZ+dR9ZiEeD9pG2SbSr69K5eDxkSCy8ZHEFgemQnBbQPeAV7nHINR4QmKQVPBRyHju+aclCHh4Q+jetWOu63gAfN4PDotMA8IxveV6LLUrsl5vRNxaZnb0nGgIPmAisubZehnD6MwTBmtgx5pByls744VIHR/AyrrQ5DdfFEz37o/lN9UhOYdDbgY3oxuEwxUIkDukbPw/3/oagDNrgjQvXdkySo3Ek00y62uPu4x8kBPS9UWMoyqT3zQ2bhxmTRdOvJ2shWXFipVTI0ssLJ/v7YSpRqaopappIMxj8vMHIX2W9PkFZPhVuqXzZ8VwOzqSF8XBcQqgTkaBc2eY3lWUfQ535SXplyQOUG+JzzII+P7RPClNLIoed5PzP+2fqvOPeYLq7VObAvxqkZmAW8poQ+YBuXQ8O+zfE9CKKfNNtXOk1Q8yzsVKFYBHXoqiGGfodpmrQusppYUy1nwWR17gzUbTKJ22mBuGQ0K7zH7ne9Lipc+Xf4vkR/lFpIWlRnG/Kzk1SowlPv085Y1lK9GJiTN5t1ptV6udXRwHM+TAA1rN2sF8dUXskfvbsj7PvC5cWr/8G79plDY5p771Z39qlFsffQDgJkejqQqfpxzFskrKdfUiG95ceU6U778vdULK1mAS1pT6LclLdkS2traMcuO2UDCnGmeg8RWcQNeQ0WmtjWM0OVDtWbAlsNaN+SzbKigxVvaY4pE3/PZtNoiNZLn2yRP9YJt9betijqW1FgeRr/a2JYmy23kE4OJ5CZR98y++ZZRPb0rZ0xfefNMoRdqMMR8Bak9hoZOxk4wxE7ZSo3VRrIVlxYqVUyP2gWXFipVTI0sMVM9TjEAn5WgCYMrKA/Vea7HLiMzCpYrY8DWCFC2mqZEwoMgsj+7+DoCDHSmmP96TXKFzZwQklpgRVmeNSIMl+IhkMt0DMXojlm8Y5OEzzaRE7rQiqy6mdO4qZV2SKsCUSarBDxAkPi5JTN47JjRVGxUOIrUUiYI4El1o2VM2JmXtxAXgaZEQL0hQ0katrJhhMYq6+VdYGtWoNY3S6YkNX2utAUg47IiTvHhJMmsmpIXQNKkmmZd9+n11ETZWZf3rlSqATlcgVZX5cScB57+LKNTtD7v8hPk4BBFrLG351X/0D43y87/8K0bRhLKXX37FKDc++QSzLUs5Pr3Y+O//+b8Y5cMPhYv5lVdfM8rVa8KvsPonAjAPD/Y5TISZihZVUl2ldQFKD/aFjWAGDosSx8tvp8XPtW1PzIuo4Qv9ATo5piISZIjJFEtd/5HwXnz3h+8Z5a03XjRK90gSvrRipn0keVjOnixLWuEV6chX9+7fw0xIbUhyizLb8e4+pFNC64c0NMSomrY+Ukq/glfHDHDWB8uiWAvLihUrp0bsA8uKFSunRpZY8iOChYjG9mgwwkzqk/Ii+AuJLRMqI9qKdW3BUhe0ePZJ6dd4aecQQLEkSHBtVSKATbI1BDtijddr3Jdoca1JnjCSPYxGZPIreAAyzS5h+xm1ltXgnEw1SqgpQnJGWqwznSy34UfsBpQXEaQCOUOaxwViq/5EJqnceJoL5qMEoFRhIlUkUTxXg1u8Rg4JwmtVpgIx5DRhbKnclDVcPX8ZwHc/EESg+V+tVtMoY22xqaEZzlZLf0psi6vT3tnZBeCQgHxtQ0BQTpf4E2Q+SCiirV/G2tqzTyTCRjjPPifpUS8xn6jKQJIi5Zeek6+effp5zOB9h3ApYbLVN77+TaPcuCWRsvNnBSmvMgx99oyc2kcfCz7yvQxApyOQp0dc7CXKPSAHWueyzESfZRtd9jnRBYwXeoimjEdnZAGJCEJTHjHWPD5l8osdAH0WMF08L11tblwXTsTDQ0F5CZFmSAK/2lh+s1caTxnl+IEsnan9ijkBpVpZWyMzJXlB1AWkzVZd8o142sxV6UbSDLMI92SxFpYVK1ZOjdgHlhUrVk6NLIGEHZawZzRoo8kUQMwsu5wXQbEhYYVyJIz67NDVF4xz1BGrcpv9vuLxBMCjtuC+aCgINF4V3DcmFfpGS2zUFiNxTaYURgPS5hG7mRiKBo+0KaOam1pi7mtDR9ZbpAwXDtkKbMDGWXOiTVgdFjBpeYrLlDmXxrYW3Kh5XGLIcnN9HUDsCmTrH7MZGgOvASRps8E6D4+xlTEJ2h02Xitkss3b338PwDvvCyTM6foIGVoM/IWsdtIFGjIcXCYkVDR98OgRgFJVPi+0BbtVSPK9KEvq7vPYWQZygQB5y9sJJ6mVRltbV41yYUPoOlzG/pgbq61CUQoKAAKukuZksogLHjNjq+wm22LtlM+uoitapMVZu14AYEpc2dkX3gufrUO7rL/RWHOFhTjxWIZtd9hB93HRYhRtmKYAKl8uLSnLZBuFhFONr2k4chwDOLorBUYTpRJkSdPmatMoh20paXqwL8rHJJ4/c5G/Tf7233j5eQAhS6b2HgkuBoPyRToWfA1gav9XXjJHyRd5juZJk+Sc7id0UrAWlhUrVk6RLM3DIj/RQGyTbqcLoMvXqVYnqPu0yNIT/UTrJ3Obhf4/ZVNdqxUB7IzknbPP7JXjjmSInFkXD2id7+/QE7tj/768OpQmOPdnuz6ASCegLx0+8kO1sGhYaWdKbc9ZY4qT75ewTHxlj/LnncmZmqJa7pvKS7hZkjM5uymnVi07AHZph47YrEVZooZkcRrQc18iD3WfVt4KU6iOyDL8V995B8DukQyrPFytGluudnlxM1l/bRSuSu5bpfVaa9TwWJoM3cDREkNKziMn/9VaFu1INAUQs+TV4QaafhWSuezqk+L9BQ3nw35bRmNNeN5D1HVnJ6l+7knu2JYlrdVYbcYq6AKXv0Ur0uNL3bBmu3QMj5nyFrIWvUvryWGPmUZNzLQJk8J6h4+wTPRHl3umlfWM47MwKk/IAk2VYu6t5ldxCiDJ2EmXPGLrLfmkHsq+jYZMcr8jk/zwUKzI47/8vlFefEYM29VGFUDKwu8tpvVlnLY+UALaXA5o6ualOWo0yl/jfU9zE9JSJFuxYuX0i31gWbFi5dTIEkioTLUHB4IBb16/iRnvrKIhNek6tHi1EalyAAyHYl5qr0e1YyvlCoC1NTIE0XNfrYqyUWGy0lTGf3R/wG3FqiwRNVSqgrbKUh5EWKGkXdwyoK9dG4X26GKPSDlUoc+10SximWifGIWEE0Kbiie7aCJbysILNyaJkiuwq9PbA7B/LG7OAfPXnKlAEoeYutsXDMIoAiL6leNMwM4hYXtvNAYwJJSbEt17ZL2aMnIyg+ZkDk9ckUS5IcGUo61cXB/AiOGR7W1xPNcbNZwg47Yg/THpp5UJw/TR7LYlznBM2miHkK1MX3gpkAnc+OgdGY0FJdrRNucUdlzMZDYleqZ6FlNByg2WbSWE2yMeeqXG24bNkEajAYBeX27FB7usaNlnjYsr98zWNQFKBV/u7bsPhcygTz7iOUkXilE0KSkn5FJqNq3RySnGFC3K/27qADhHF3vKlL1Of/z4Hqhq32LGWPq8vg+74nS/PGKRXHMdgM/DuEqRTEY5ZSjxmTaYOepZZ47Y3wb9LCS0YsXKT4PYB5YVK1ZOjSyBhNFUSw3kk+PjNoCAbGGanlNmUxYNBUYsKHFon9eaYpSC3RPHTK7pDaYAWi0BRyXmGdVrsm8rZJ19qlPS9jDgHAQJhgShyWQMwGfYokDb1+W+MZFOxjNFXrXDE2FCWfEExrUkD3URt0ZkUygxdsaErNSfhy39RJDyoNcDMCJw9hh/ySBmuWaNOZybl+fasBCKCWsZj2h6uBbZSKZODmuPAHDKuKT2wKmtSLJbnxlzfRLjalzYxLA0qKr9XzV7aFGuv/vXRkmI0Vwm5kzGQwCTMct6xuQLZFyswlFr5HQcdgWE6rLHE+bQcVYGekyYyKZdOZWTo8rksIwJWZ5OiTHZBo/dZLhwHE8AdFg25LHl7bvvvCuzbcit3jy7ZZSDA8lsuvERmamnyyFhlvMu8COFhGRrcB0t0qI7gmeUsfxIT9ZxUzxGC671N3LKEUPMHj0YdbJH9pXogkBVHQiOHwLwGK/UnsT5/JWaXHOpPA2zEsny55bp/CVKyF/oQn2SirWwrFixcmrEPrCsWLFyamRZI1WS5Gkk7pXPvIgZ3oJUe3Qo7UGslQGa0iZ/tSC7wCBdic1Ws8IYQFhg8C5gx1OPliFHc8lrrqGHcok5nzQ8p0RVJssxb2vKKBKI8rRqR0kaHKLdSs5Ip71elkPCkK19qkwOTBMx3ccMF3oMQq1tbHC2emoCSyvlMmZab2ptReqzoEFT8rjIgXb91MVmbGuDDW+ubm0CeLB/aP6tV5syPKk4XJfx0IbsEoTkHSTBYeqx/IWXpl6vAoi4bpW67FIqLV8lAMODB5wlV1vbZGYJgIA34dkVWfzVCkt/tJssM28TBvg8oonRQCnrGJ/yPACOEkhwECX1X+Et4RG2qIchY4ZtsymTWV+XTMuHhx0AD/eF5ODlt94wCsuicMAI4MGekEDssLVqiaHGp68xA/Ybb2NGlP1dZ+uoK8NR3KTrNs9qsEiEkbkOgIxwMqf+YDuoIe80dTVcZLsslwdSnKd0KVkSAcjyDrvzWdk6t9SZd8Vozq3iVs2szpDh5L5Ks2ItLCtWrJwasQ8sK1asnBpZYskrqnMJCTfPbQJot7UwTdBQoy5xpZ1dyZ3rkaRBbdAG8zkd8hNoALHZdAD4OafdfNGZlu9rhDEhmhgw/22qQTRoDMIBEBJSpUQTXkHbZNHi5ZELxD4BdC+GlpYtEYA6EwtLFe2syVxZwi7NxDt3ecsovbZgtM6xzL/klQAEFZl8maukJBNqJXtcU8IL1IjmylWpTHSZqfjsM78BYO9IcjKnrN8cMvA3JB1jSHykSLlA3D3oS6rnmFHCMAwBHLPozPVlSrW6DPIH/+3reFzikRwx0ppBhuRMBNbl7bRSlUEukU4v1jCrImaGoRXOK0O/r4Wb6RRAwg0KmspIIOMxAzMM5AL5gRw64d2ysiokfBvrAufbgymAARewUSPx5LUrcoIsvVSqxatba5y2fFUqaNL1Y6LxMjUhNHlSs0O1u5cy+TkLRXnOYrTx8f983lBBoDFH+WqzJqe8slqbO2KFHhjfhCM5pZh5yDPJ5HpK88fWuWmZpPbZmyd1xDzmVbEWlhUrVk6NLDEf1ljPPWHByt7DfQADZu7oO63Dogpt7F5mLlVeWEBFa2jU+26eqpowpalPE3aUSflujPle1eSaESdTpAvf44FMmb7yE+hxS2W+Tlma4+hbekR7iqU5+Ymc4AgMQy4drb8Se757ZO91QvlEE5fUQoy0z1ChBODSplTDVMqy7/GhpPBoX1uPZ+TrW4YvKG04OuYbb+3sWQDPPCsvfyWDznKfLp2jmhSTc0vQn61FQBSz+1iJN5jQ5NHU+tf4j3O7KGmvu5A9NJA4iRxFAyk/88bnjPLUM6/I6azJSg5GbaNo59aU5FDawF1KWHg/jHjzuMycGkznm+Devrctn0zkvuonTOvjbyQMygAmzH2bcnGqVZlAhaTAai/UAjFVcnPphH4wGdSe4mwfpw8G/eiYdXDr1fzbyIXVhe8xx62gRGI0sYplTbTkiXC2ahQbA01ZrZW2QSW/yksm4cxt4/EHPo2NUTzPArIo1sKyYsXKqRH7wLJixcqpkSWQsH0otKeDriC+9tExAJcZFgVmLc0Yq4RU2s9D+9DQTk66ZHQgPDEpSGO6qDUfX/cNyQsYEmSVWFleq4ubWV25WrUTGuhEj3pKi3fI7iNePjXln+A60Fg96Iw57PI2J0WlfCChcFBjp9hQgEB7KGd08/odOUetzCBO8ZwAQMBkNGWbW2mIT1eRoDLVVZmmNGoLf4NHE1orS/b39gCUalJkr+BaO+Jo5lfAkiYdRJdUs5OUwNe4S0tlWa6EHYm0M82i+FxkZdbWxCK3EABIIjncgC4ILZCaktn3xg2BbN/+tjj1xwyPKBmA8mf4foCZOo8h4zPquZ+SpUNb43zjm38p585hP70v4ZHDHrPqPB9AkYTLMUG9x8vhaOegHNaBc5NFSE7IV8tyAEiEzltR0eLfBQAu8ByoZ0bxsnxRoJt8xAZRY/4wA3ZFcjkZ7/FQQH4MDqtoTieg9Mf5qS0oM/N+7LyyhUQzFWthWbFi5dSIfWBZsWLl1MiyrjmM/WlOfau5AqBUJkkb8cWI6Tl37kkNvVZoF0NWP3BjBXoTbcgaRZjJeMoWaL2SvBfm/INVTdARu5j4LBxv1JoAiuU6R2P0cDrP6DYiLMpYxqHYJ+XKuCc80stsHKLkFvu7Eter1gUtDsdyRsdMaAqZC6YVRQdHHQAeU94aZBlXuBoSgCsRQ0QMpTxqEcN25Yog5cy0eGEsLVsICU0nMoMeoXqRjAsaYJ2yoH8SycaGBNElQkjIxZhmy4NfmCl/yQj04pwOJAWQMWvpnXd/aJT9fQnV1Vj6UyEvyOYq+ydtSCxbqSLVpWBSzLTPwJAAcMAbr9tuG2XrkuRJXb78ihyxKbfNjbtSVfOjDz+SM8kyAHXGmissySoSEiakIdGUQM/TXCdNblwOdtT7MVHSRy0byuGYIqb5Op5sIQAnx104mtIBelx2TVjrHsuyh3SahLxtQrojzC3nsnguvxMWIGF+RCpLsKGWH7kuZhwLP6FGx1pYVqxYOTViH1hWrFg5NbIEEmpfLLXLgrAAYGdPCJ6Vpj1j4qKnJRS0Y7URaUJsmBN6KVkaEgBBUcsj2FiJikYhnbz3kTI6EOjR4lQO7/ZRB0DQI0sfT2dJBDMPMLH6nLAoL1w4ISKjjTw1xTQmy3iv05YjssFRmXA14V5TZnjGaQRgwFBpleUpYGA0dQUsjIZisWv5SImRSo+NvxI2JTNWd85LRxSmDPQjksdrC6+UxnuR5UFaGVNg9Uzq+AASrb/3tKxKcybnpXskoUxtKxvzSvV7PQAxP3/9jTdl8rzu2p+1we5kF84J5i2WeO6MdmU5i2SEmb6/PpnvC7y4ffKyq5RYRKUg6oWrTxjl2ScvzW7p85RzzkJ2ZkvIrK/gtEhuPJeWgeadzom2jE0XCPBmwDYh4dz/M6KxcpMY7CykceqvO81RpLZllVsi1kgiKQPTPOZuRluMYM5PSSuK3Lwp2QJa5FeOm83ukthGqlasWLFixYoVK1asWLFixYoVK1asWLFixYoVK1asWLFixYqVUyf/D3PcGe48X+nJAAAAAElFTkSuQmCC\n",
  1208. "text/plain": [
  1209. "<PIL.Image.Image image mode=RGB size=400x100 at 0x7F1EC53EAB38>"
  1210. ]
  1211. },
  1212. "execution_count": 6,
  1213. "metadata": {},
  1214. "output_type": "execute_result"
  1215. }
  1216. ],
  1217. "source": [
  1218. "dataiter = iter(trainloader)\n",
  1219. "images, labels = dataiter.next() # 返回4张图片及标签\n",
  1220. "print(' '.join('%11s'%classes[labels[j]] for j in range(4)))\n",
  1221. "show(tv.utils.make_grid((images+1)/2)).resize((400,100))"
  1222. ]
  1223. },
  1224. {
  1225. "cell_type": "markdown",
  1226. "metadata": {},
  1227. "source": [
  1228. "#### 定义网络\n",
  1229. "\n",
  1230. "拷贝上面的LeNet网络,修改self.conv1第一个参数为3通道,因CIFAR-10是3通道彩图。"
  1231. ]
  1232. },
  1233. {
  1234. "cell_type": "code",
  1235. "execution_count": 7,
  1236. "metadata": {},
  1237. "outputs": [
  1238. {
  1239. "name": "stdout",
  1240. "output_type": "stream",
  1241. "text": [
  1242. "Net(\n",
  1243. " (conv1): Conv2d(3, 6, kernel_size=(5, 5), stride=(1, 1))\n",
  1244. " (conv2): Conv2d(6, 16, kernel_size=(5, 5), stride=(1, 1))\n",
  1245. " (fc1): Linear(in_features=400, out_features=120, bias=True)\n",
  1246. " (fc2): Linear(in_features=120, out_features=84, bias=True)\n",
  1247. " (fc3): Linear(in_features=84, out_features=10, bias=True)\n",
  1248. ")\n"
  1249. ]
  1250. }
  1251. ],
  1252. "source": [
  1253. "import torch.nn as nn\n",
  1254. "import torch.nn.functional as F\n",
  1255. "\n",
  1256. "class Net(nn.Module):\n",
  1257. " def __init__(self):\n",
  1258. " super(Net, self).__init__()\n",
  1259. " self.conv1 = nn.Conv2d(3, 6, 5) \n",
  1260. " self.conv2 = nn.Conv2d(6, 16, 5) \n",
  1261. " self.fc1 = nn.Linear(16*5*5, 120) \n",
  1262. " self.fc2 = nn.Linear(120, 84)\n",
  1263. " self.fc3 = nn.Linear(84, 10)\n",
  1264. "\n",
  1265. " def forward(self, x): \n",
  1266. " x = F.max_pool2d(F.relu(self.conv1(x)), (2, 2)) \n",
  1267. " x = F.max_pool2d(F.relu(self.conv2(x)), 2) \n",
  1268. " x = x.view(x.size()[0], -1) \n",
  1269. " x = F.relu(self.fc1(x))\n",
  1270. " x = F.relu(self.fc2(x))\n",
  1271. " x = self.fc3(x) \n",
  1272. " return x\n",
  1273. "\n",
  1274. "\n",
  1275. "net = Net()\n",
  1276. "print(net)"
  1277. ]
  1278. },
  1279. {
  1280. "cell_type": "markdown",
  1281. "metadata": {},
  1282. "source": [
  1283. "#### 定义损失函数和优化器(loss和optimizer)"
  1284. ]
  1285. },
  1286. {
  1287. "cell_type": "code",
  1288. "execution_count": 8,
  1289. "metadata": {},
  1290. "outputs": [],
  1291. "source": [
  1292. "from torch import optim\n",
  1293. "criterion = nn.CrossEntropyLoss() # 交叉熵损失函数\n",
  1294. "optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9)"
  1295. ]
  1296. },
  1297. {
  1298. "cell_type": "markdown",
  1299. "metadata": {},
  1300. "source": [
  1301. "### 训练网络\n",
  1302. "\n",
  1303. "所有网络的训练流程都是类似的,不断地执行如下流程:\n",
  1304. "\n",
  1305. "- 输入数据\n",
  1306. "- 前向传播+反向传播\n",
  1307. "- 更新参数\n"
  1308. ]
  1309. },
  1310. {
  1311. "cell_type": "code",
  1312. "execution_count": 10,
  1313. "metadata": {},
  1314. "outputs": [
  1315. {
  1316. "name": "stderr",
  1317. "output_type": "stream",
  1318. "text": [
  1319. "/home/bushuhui/.virtualenv/dl/lib/python3.5/site-packages/ipykernel_launcher.py:25: UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number\n"
  1320. ]
  1321. },
  1322. {
  1323. "name": "stdout",
  1324. "output_type": "stream",
  1325. "text": [
  1326. "[1, 2000] loss: 2.210\n",
  1327. "[1, 4000] loss: 1.958\n",
  1328. "[1, 6000] loss: 1.723\n",
  1329. "[1, 8000] loss: 1.590\n",
  1330. "[1, 10000] loss: 1.532\n",
  1331. "[1, 12000] loss: 1.467\n",
  1332. "[2, 2000] loss: 1.408\n",
  1333. "[2, 4000] loss: 1.374\n",
  1334. "[2, 6000] loss: 1.345\n",
  1335. "[2, 8000] loss: 1.331\n",
  1336. "[2, 10000] loss: 1.338\n",
  1337. "[2, 12000] loss: 1.286\n",
  1338. "Finished Training\n"
  1339. ]
  1340. }
  1341. ],
  1342. "source": [
  1343. "from torch.autograd import Variable\n",
  1344. "\n",
  1345. "t.set_num_threads(8)\n",
  1346. "for epoch in range(2): \n",
  1347. " \n",
  1348. " running_loss = 0.0\n",
  1349. " for i, data in enumerate(trainloader, 0):\n",
  1350. " \n",
  1351. " # 输入数据\n",
  1352. " inputs, labels = data\n",
  1353. " inputs, labels = Variable(inputs), Variable(labels)\n",
  1354. " \n",
  1355. " # 梯度清零\n",
  1356. " optimizer.zero_grad()\n",
  1357. " \n",
  1358. " # forward + backward \n",
  1359. " outputs = net(inputs)\n",
  1360. " loss = criterion(outputs, labels)\n",
  1361. " loss.backward() \n",
  1362. " \n",
  1363. " # 更新参数 \n",
  1364. " optimizer.step()\n",
  1365. " \n",
  1366. " # 打印log信息\n",
  1367. " running_loss += loss.data[0]\n",
  1368. " if i % 2000 == 1999: # 每2000个batch打印一下训练状态\n",
  1369. " print('[%d, %5d] loss: %.3f' \\\n",
  1370. " % (epoch+1, i+1, running_loss / 2000))\n",
  1371. " running_loss = 0.0\n",
  1372. "print('Finished Training')"
  1373. ]
  1374. },
  1375. {
  1376. "cell_type": "markdown",
  1377. "metadata": {},
  1378. "source": [
  1379. "此处仅训练了2个epoch(遍历完一遍数据集称为一个epoch),来看看网络有没有效果。将测试图片输入到网络中,计算它的label,然后与实际的label进行比较。"
  1380. ]
  1381. },
  1382. {
  1383. "cell_type": "code",
  1384. "execution_count": null,
  1385. "metadata": {
  1386. "lines_to_next_cell": 2
  1387. },
  1388. "outputs": [],
  1389. "source": [
  1390. "dataiter = iter(testloader)\n",
  1391. "images, labels = dataiter.next() # 一个batch返回4张图片\n",
  1392. "print('实际的label: ', ' '.join(\\\n",
  1393. " '%08s'%classes[labels[j]] for j in range(4)))\n",
  1394. "show(tv.utils.make_grid(images / 2 - 0.5)).resize((400,100))"
  1395. ]
  1396. },
  1397. {
  1398. "cell_type": "markdown",
  1399. "metadata": {},
  1400. "source": [
  1401. "接着计算网络预测的label:"
  1402. ]
  1403. },
  1404. {
  1405. "cell_type": "code",
  1406. "execution_count": 12,
  1407. "metadata": {},
  1408. "outputs": [
  1409. {
  1410. "name": "stdout",
  1411. "output_type": "stream",
  1412. "text": [
  1413. "预测结果: cat ship ship ship\n"
  1414. ]
  1415. }
  1416. ],
  1417. "source": [
  1418. "# 计算图片在每个类别上的分数\n",
  1419. "outputs = net(Variable(images))\n",
  1420. "# 得分最高的那个类\n",
  1421. "_, predicted = t.max(outputs.data, 1)\n",
  1422. "\n",
  1423. "print('预测结果: ', ' '.join('%5s'\\\n",
  1424. " % classes[predicted[j]] for j in range(4)))"
  1425. ]
  1426. },
  1427. {
  1428. "cell_type": "markdown",
  1429. "metadata": {},
  1430. "source": [
  1431. "已经可以看出效果,准确率50%,但这只是一部分的图片,再来看看在整个测试集上的效果。"
  1432. ]
  1433. },
  1434. {
  1435. "cell_type": "code",
  1436. "execution_count": 13,
  1437. "metadata": {},
  1438. "outputs": [
  1439. {
  1440. "name": "stdout",
  1441. "output_type": "stream",
  1442. "text": [
  1443. "10000张测试集中的准确率为: 54 %\n"
  1444. ]
  1445. }
  1446. ],
  1447. "source": [
  1448. "correct = 0 # 预测正确的图片数\n",
  1449. "total = 0 # 总共的图片数\n",
  1450. "for data in testloader:\n",
  1451. " images, labels = data\n",
  1452. " outputs = net(Variable(images))\n",
  1453. " _, predicted = t.max(outputs.data, 1)\n",
  1454. " total += labels.size(0)\n",
  1455. " correct += (predicted == labels).sum()\n",
  1456. "\n",
  1457. "print('10000张测试集中的准确率为: %d %%' % (100 * correct / total))"
  1458. ]
  1459. },
  1460. {
  1461. "cell_type": "markdown",
  1462. "metadata": {},
  1463. "source": [
  1464. "训练的准确率远比随机猜测(准确率10%)好,证明网络确实学到了东西。"
  1465. ]
  1466. },
  1467. {
  1468. "cell_type": "markdown",
  1469. "metadata": {},
  1470. "source": [
  1471. "#### 在GPU训练\n",
  1472. "就像之前把Tensor从CPU转到GPU一样,模型也可以类似地从CPU转到GPU。"
  1473. ]
  1474. },
  1475. {
  1476. "cell_type": "code",
  1477. "execution_count": 44,
  1478. "metadata": {},
  1479. "outputs": [],
  1480. "source": [
  1481. "if t.cuda.is_available():\n",
  1482. " net.cuda()\n",
  1483. " images = images.cuda()\n",
  1484. " labels = labels.cuda()\n",
  1485. " output = net(Variable(images))\n",
  1486. " loss= criterion(output,Variable(labels))"
  1487. ]
  1488. },
  1489. {
  1490. "cell_type": "markdown",
  1491. "metadata": {},
  1492. "source": [
  1493. "如果发现在GPU上并没有比CPU提速很多,实际上是因为网络比较小,GPU没有完全发挥自己的真正实力。"
  1494. ]
  1495. },
  1496. {
  1497. "cell_type": "markdown",
  1498. "metadata": {},
  1499. "source": [
  1500. "对PyTorch的基础介绍至此结束。总结一下,本节主要包含以下内容。\n",
  1501. "\n",
  1502. "1. Tensor: 类似Numpy数组的数据结构,与Numpy接口类似,可方便地互相转换。\n",
  1503. "2. autograd/Variable: Variable封装了Tensor,并提供自动求导功能。\n",
  1504. "3. nn: 专门为神经网络设计的接口,提供了很多有用的功能(神经网络层,损失函数,优化器等)。\n",
  1505. "4. 神经网络训练: 以CIFAR-10分类为例演示了神经网络的训练流程,包括数据加载、网络搭建、训练及测试。\n",
  1506. "\n",
  1507. "通过本节的学习,相信读者可以体会出PyTorch具有接口简单、使用灵活等特点。从下一章开始,本书将深入系统地讲解PyTorch的各部分知识。"
  1508. ]
  1509. }
  1510. ],
  1511. "metadata": {
  1512. "kernelspec": {
  1513. "display_name": "Python 3",
  1514. "language": "python",
  1515. "name": "python3"
  1516. },
  1517. "language_info": {
  1518. "codemirror_mode": {
  1519. "name": "ipython",
  1520. "version": 3
  1521. },
  1522. "file_extension": ".py",
  1523. "mimetype": "text/x-python",
  1524. "name": "python",
  1525. "nbconvert_exporter": "python",
  1526. "pygments_lexer": "ipython3",
  1527. "version": "3.5.2"
  1528. }
  1529. },
  1530. "nbformat": 4,
  1531. "nbformat_minor": 2
  1532. }

机器学习越来越多应用到飞行器、机器人等领域,其目的是利用计算机实现类似人类的智能,从而实现装备的智能化与无人化。本课程旨在引导学生掌握机器学习的基本知识、典型方法与技术,通过具体的应用案例激发学生对该学科的兴趣,鼓励学生能够从人工智能的角度来分析、解决飞行器、机器人所面临的问题和挑战。本课程主要内容包括Python编程基础,机器学习模型,无监督学习、监督学习、深度学习基础知识与实现,并学习如何利用机器学习解决实际问题,从而全面提升自我的《综合能力》。