36b018daf15ae8af90cf982fd01bff2008ced3bd,examples/experimental/simple_cifar.py,,,#,99

Before Change

    epochs=5,
    loaders=get_loaders(),
    criterion=CrossEntropyLoss(),
    optimizer=Adam(model.parameters(), lr=0.001),
    scheduler=None,
    callbacks=OrderedDict(
        tqdm=VerboseCallback(),

After Change

model = SimpleNet()
criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters(), lr=0.005)
scheduler = torch.optim.lr_scheduler.MultiStepLR(
    optimizer,
    milestones=[2, 3],
    gamma=0.2)
runner = SupervisedRunner(model=model, input_key="image")

// training
runner.train(
    logdir="./logs/01",
    epochs=5,
    loaders=get_loaders(),
    criterion=criterion,
    optimizer=optimizer,
    scheduler=scheduler,
    callbacks=OrderedDict(
        tqdm=VerboseCallback(),
        loss=LossCallback(),

Instances