f713e25a706eea6866a3a4324d5bf5fcada5dbcd,examples/gluon_simple.py,,,#,17

Before Change

    "--epochs", type=int, default=10, help="number of epochs to train (default: 10)"
)
parser.add_argument("--cuda", action="store_true", default=False, help="Train on GPU with CUDA")
parser.add_argument(
    "--log-interval",
    type=int,
    default=100,
    metavar="N",
    help="how many batches to wait before logging training status",
)
opt = parser.parse_args()


// define network

After Change

CUDA = False
EPOCHS = 10
BATCHSIZE = 128
DEVICE = torch.device("cpu")
LOG_INTERVAL = 100


// define network

Instances