python - the_model = TheModelClass(*args, **kwargs) 是什么意思？

Question

我正在使用 PyTorch 进行图像分类。训练最多后，我想保存训练好的模型。

我不明白什么意思

the_model = TheModelClass(*args, **kwargs)

这行代码由 PyTorch 网站 (https://pytorch.org/docs/master/notes/serialization.html) 给出。

Answer 1

这个问题是the_model = TheModelClass(*args, **kwargs)意味着您必须首先定义一个 ModelClass 对象。然后你可以使用模型对象来加载磁盘顺序对象。例如:

in_feats = data.x.shape[1]
n_hidden = params["n_hidden"]
n_classes = 2
best_model = OwnGCN(in_c=in_feats, hid_c=n_hidden, out_c=n_classes)
best_model.load_state_dict(torch.load(PATH))