61421e8d341312e02dff23b46acde4261808dab3,train.py,,,#,15

Before Change

parser.add_argument_group("Optimization related arguments")
parser.add_argument("-batch_size", default=4, help="Batch size")
parser.add_argument("-learning_rate", default=1e-3, help="Learning rate")
parser.add_argument("-dropout", default=0.5, help="Dropout")
parser.add_argument("-num_epochs", default=20, help="Epochs")
parser.add_argument("-weight_init", default="xavier", choices=["xavier", "kaiming"],
                        help="Weight initialization strategy")
parser.add_argument("-gpuid", default=0, help="GPU id to use")

After Change

args = parser.parse_args()
for arg in vars(args):
    print("{:<20}: {}".format(arg, getattr(args, arg)))

// seed for reproducibility
torch.manual_seed(1234)

Instances