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