10e28a98e185571218b1dcb9f9b91f663a8ec673,catalyst/dl/scripts/trace.py,,trace_model_from_checkpoint,#Any#Any#,14

Before Change

    print("Load model state from checkpoints/best.pth")
    model = experiment.get_model(next(iter(experiment.stages)))
    model.load_state_dict(torch.load(checkpoint_path)["model_state_dict"])

    print("Tracing")
    traced = trace_model(model, experiment, RunnerType, method_name)

After Change

    print("Load model state from checkpoints/best.pth")
    model = experiment.get_model(next(iter(experiment.stages)))
    checkpoint = UtilsFactory.load_checkpoint(checkpoint_path)
    UtilsFactory.unpack_checkpoint(checkpoint, model=model)

    print("Tracing")
    traced = trace_model(model, experiment, RunnerType, method_name)

Instances