molprop.utils.transfer_learning.load_pretrained_params

molprop.utils.transfer_learning.load_pretrained_params(model, pre_model_path)

Loads pretrained parameters to model.

Parameters:

• model (torch.nn.Module) – Model used in training.

• pre_model_path (str) – Path of the pretrained model (if existent).

Returns:

Model to which the parameters are loaded. Returns original model, if error occurs.

Return type:

torch.nn.Module