Skip to content

Commit 6b1cef4

Browse files
committed
Fix w2d2 tut1 errors
1 parent 7f5918a commit 6b1cef4

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

tutorials/W2D2_Convnets/W2D2_Tutorial1.ipynb

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -131,12 +131,14 @@
131131
"import scipy.signal\n",
132132
"import numpy as np\n",
133133
"import matplotlib.pyplot as plt\n",
134+
"import IPython\n",
134135
"\n",
135136
"import torch.nn as nn\n",
136137
"import torch.nn.functional as F\n",
137138
"\n",
138139
"import torchvision.transforms as transforms\n",
139140
"import torchvision.datasets as datasets\n",
141+
"from torchvision.utils import make_grid\n",
140142
"from torch.utils.data import DataLoader\n",
141143
"\n",
142144
"from tqdm.notebook import tqdm, trange\n",
@@ -4093,7 +4095,7 @@
40934095
"source": [
40944096
"# @title Finetune ResNet\n",
40954097
"pretrained_accs = []\n",
4096-
"for epoch in tqdm.tqdm(range(10)):\n",
4098+
"for epoch in tqdm(range(10)):\n",
40974099
" # Train loop\n",
40984100
" for batch in pokemon_train_loader:\n",
40994101
" images, labels = batch\n",
@@ -4176,7 +4178,7 @@
41764178
"source": [
41774179
"# @title Finetune readout of ResNet\n",
41784180
"linreadout_accs = []\n",
4179-
"for epoch in tqdm.tqdm(range(10)):\n",
4181+
"for epoch in tqdm(range(10)):\n",
41804182
" # Train loop\n",
41814183
" for batch in pokemon_train_loader:\n",
41824184
" images, labels = batch\n",
@@ -4258,7 +4260,7 @@
42584260
"source": [
42594261
"# @title Train ResNet from scratch\n",
42604262
"scratch_accs = []\n",
4261-
"for epoch in tqdm.tqdm(range(10)):\n",
4263+
"for epoch in tqdm(range(10)):\n",
42624264
" # Train loop\n",
42634265
" for batch in pokemon_train_loader:\n",
42644266
" images, labels = batch\n",

0 commit comments

Comments
 (0)