Import mlr3torch Checkpoint

Loads a torch checkpoint into a trained learner for inference, or attaches it for potential fine-tuning if the learner is not yet trained.

Usage

import_mlr3torch_checkpoint(learner, path, strict = TRUE)

Arguments

learner

mlr3 learner or GraphLearner.

path

Checkpoint path created by export_mlr3torch_checkpoint().

strict

Logical, enforce exact layer matching.

Value

The updated learner.