The code I use is here:

  
      
      GitHub
  
  
    

zhunzhong07/Random-Erasing 8

Random-Erasing - Random Erasing Data Augmentation


  
  
    
    
  
  


I just modified the DataLoader to an ImageLoader and provided a path to locally hosted Mnist.