File tree Expand file tree Collapse file tree 2 files changed +7
-3
lines changed
02-intermediate/language_model
03-advanced/generative_adversarial_network Expand file tree Collapse file tree 2 files changed +7
-3
lines changed Original file line number Diff line number Diff 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 ()
Original file line number Diff line number Diff line change 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))])
2529transform = 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
3135mnist = torchvision .datasets .MNIST (root = '../../data/' ,
You can’t perform that action at this time.
0 commit comments