2775c15702fd52f081ff8fded51f4f38877659df,examples/_tests_scripts/z_classification.py,,,#,87

Before Change

// In[ ]:

from catalyst.dl.callbacks import InferCallback
loaders = collections.OrderedDict([("infer", loaders["train"])])
runner.infer(
    model=model, loaders=loaders, callbacks=[InferCallback()], check=True
)

After Change

runner = SupervisedRunner()

// model training
runner.train(
    model=model,
    criterion=criterion,
    optimizer=optimizer,
    scheduler=scheduler,
    loaders={"train": loaders["train"]},
    valid_loader="train",
    callbacks=[
        AccuracyCallback(accuracy_args=[1, 3, 5]),
    ],
    logdir=logdir,
    num_epochs=num_epochs,
    check=True
)

// In[ ]:

// utils.plot_metrics(logdir=logdir, step="epoch", metrics=["loss", "accuracy01"])

Instances