File tree 2 files changed +7
-3
lines changed
02-intermediate/language_model
03-advanced/generative_adversarial_network
2 files changed +7
-3
lines changed Original file line number Diff line number Diff line change @@ -76,7 +76,7 @@ def detach(states):
76
76
loss = criterion (outputs , targets .reshape (- 1 ))
77
77
78
78
# Backward and optimize
79
- model .zero_grad ()
79
+ optimizer .zero_grad ()
80
80
loss .backward ()
81
81
clip_grad_norm_ (model .parameters (), 0.5 )
82
82
optimizer .step ()
Original file line number Diff line number Diff line change 22
22
os .makedirs (sample_dir )
23
23
24
24
# 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))])
25
29
transform = transforms .Compose ([
26
30
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 ] )])
29
33
30
34
# MNIST dataset
31
35
mnist = torchvision .datasets .MNIST (root = '../../data/' ,
You can’t perform that action at this time.
0 commit comments