chemprop.utils.v1_to_v2
#
Module Contents#
Functions#
|
Converts v1 model dictionary to a v2 state dictionary |
|
Converts v1 model dictionary to v2 hyper_parameters dictionary |
|
Converts a v1 model dictionary from a loaded .pt file to a v2 model dictionary |
|
Converts a v1 model .pt file to a v2 model .ckpt file |
- chemprop.utils.v1_to_v2.convert_state_dict_v1_to_v2(model_v1_dict)[source]#
Converts v1 model dictionary to a v2 state dictionary
- Parameters:
model_v1_dict (dict)
- Return type:
dict
- chemprop.utils.v1_to_v2.convert_hyper_parameters_v1_to_v2(model_v1_dict)[source]#
Converts v1 model dictionary to v2 hyper_parameters dictionary
- Parameters:
model_v1_dict (dict)
- Return type:
dict