#1 更新 'MNIST/main.py'

Merged
zonghuia merged 1 commits from zonghuia-patch-1 into master 1 year ago
  1. +1
    -1
      MNIST/main.py

+ 1
- 1
MNIST/main.py View File

@@ -97,7 +97,7 @@ if __name__=='__main__':
help='input batch size for training (default: 128)')
parser.add_argument('--test-batch-size', type=int, default=128, metavar='N',
help='input batch size for testing (default: 128)')
parser.add_argument('--epochs', type=int, default=60, metavar='N',
parser.add_argument('--epochs', type=int, default=200, metavar='N',
help='number of epochs to train (default: 60)')
parser.add_argument('--lr-epochs', type=int, default=15, metavar='N',
help='number of epochs to decay the lr (default: 15)')


Loading…
Cancel
Save