MultiModalPredictor.load

classmethod MultiModalPredictor.load(path: str, resume: bool | None = False, verbosity: int | None = 3)[source]

从指定目录 path 加载预测器对象。待加载的预测器可以是完全或部分通过 .fit() 训练的。如果之前的训练已完成,它将加载 checkpoint model.ckpt。否则,如果之前的训练意外中断,可以通过设置 resume=True 加载 last.ckpt checkpoint。它也支持通过给定路径加载特定的 checkpoint。

警告

autogluon.multimodal.MultiModalPredictor.load() 隐式使用了 pickle 模块,已知该模块不安全。可以构造恶意 pickle 数据,在反序列化时执行任意代码。切勿加载可能来自不受信任来源或可能已被篡改的数据。仅加载您信任的数据。

参数:
  • path – 加载预测器对象的目录。

  • resume – 是否从 last.ckpt 恢复训练。这在训练意外中断时非常有用,我们可以从上次保存的 checkpoint 恢复训练。

  • verbosity – 详细程度级别范围从 0 到 4,控制打印信息的多少。级别越高,打印的详细信息越多。您可以设置 verbosity = 0 来抑制警告。

返回类型:

加载的预测器对象。