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

// In[ ]:

runner_out = runner.predict_loader(
    loaders["valid"], resume=f"{logdir}/checkpoints/best.pth"
)

// In[ ]:

runner_out.shape

Instances