Please review my version.

  
      github.com
  
  
    aurotripathy/menace/blob/master/mnist_multigpu_hogwild/main.py 25
"""
Adding multi-gpu support to mnist w/hogwild
"""
from __future__ import print_function
import argparse
import torch
import torch.multiprocessing as mp
from model import Net
from train import train, test
from shared_optim import SharedAdam


# Training settings
parser = argparse.ArgumentParser(description='PyTorch MNIST Example')
parser.add_argument('--batch-size', type=int, default=64, metavar='N',
                    help='input batch size for training (default: 64)')
parser.add_argument('--test-batch-size', type=int, default=1000, metavar='N',
                    help='input batch size for testing (default: 1000)')
parser.add_argument('--epochs', type=int, default=10, metavar='N',
                    help='number of epochs to train (default: 10)')


  This file has been truncated. show original

  
  
    
    
  
  


I’m happy to fix issues, improve readability.
This really is derived from an RL implementation by @dgriff available here 4