Skip to content

Commit dd297a1

Browse files
committed
fix yunjey#187 Typos in language model line 79 and generative_adversarial_network line 25-28
1 parent 56bba93 commit dd297a1

File tree

2 files changed

+7
-3
lines changed
  • tutorials
    • 02-intermediate/language_model
    • 03-advanced/generative_adversarial_network

2 files changed

+7
-3
lines changed

tutorials/02-intermediate/language_model/main.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,7 @@ def detach(states):
7676
loss = criterion(outputs, targets.reshape(-1))
7777

7878
# Backward and optimize
79-
model.zero_grad()
79+
optimizer.zero_grad()
8080
loss.backward()
8181
clip_grad_norm_(model.parameters(), 0.5)
8282
optimizer.step()

tutorials/03-advanced/generative_adversarial_network/main.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,10 +22,14 @@
2222
os.makedirs(sample_dir)
2323

2424
# Image processing
25+
# transform = transforms.Compose([
26+
# transforms.ToTensor(),
27+
# transforms.Normalize(mean=(0.5, 0.5, 0.5), # 3 for RGB channels
28+
# std=(0.5, 0.5, 0.5))])
2529
transform = transforms.Compose([
2630
transforms.ToTensor(),
27-
transforms.Normalize(mean=(0.5, 0.5, 0.5), # 3 for RGB channels
28-
std=(0.5, 0.5, 0.5))])
31+
transforms.Normalize(mean=[0.5], # 1 for greyscale channels
32+
std=[0.5])])
2933

3034
# MNIST dataset
3135
mnist = torchvision.datasets.MNIST(root='../../data/',

0 commit comments

Comments
 (0)