algorithm/pytorch-tutorial
1
0
Fork 0
mirror of https://github.com/yunjey/pytorch-tutorial.git synced 2025-07-06 01:15:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix #187 Typos in language model line 79 and generative_adversarial_network line 25-28

Browse Source
This commit is contained in:
qy-yang
2019-08-27 17:31:42 +08:00
parent 56bba93619
commit dd297a1307
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tutorials/02-intermediate/language_model/main.py
View File

@ -76,7 +76,7 @@ for epoch in range(num_epochs):
loss = criterion(outputs, targets.reshape(-1)) loss = criterion(outputs, targets.reshape(-1))
# Backward and optimize # Backward and optimize
model.zero_grad() optimizer.zero_grad()
loss.backward() loss.backward()
clip_grad_norm_(model.parameters(), 0.5) clip_grad_norm_(model.parameters(), 0.5)
optimizer.step() optimizer.step()

8
tutorials/03-advanced/generative_adversarial_network/main.py
View File

@ -22,10 +22,14 @@ if not os.path.exists(sample_dir):
os.makedirs(sample_dir) os.makedirs(sample_dir)
# Image processing # Image processing
# transform = transforms.Compose([
# transforms.ToTensor(),
# transforms.Normalize(mean=(0.5, 0.5, 0.5), # 3 for RGB channels
# std=(0.5, 0.5, 0.5))])
transform = transforms.Compose([ transform = transforms.Compose([
transforms.ToTensor(), transforms.ToTensor(),
transforms.Normalize(mean=(0.5, 0.5, 0.5), # 3 for RGB channels transforms.Normalize(mean=[0.5], # 1 for greyscale channels
std=(0.5, 0.5, 0.5))]) std=[0.5])])
# MNIST dataset # MNIST dataset
mnist = torchvision.datasets.MNIST(root='../../data/', mnist = torchvision.datasets.MNIST(root='../../data/',