gigl.src.common.utils.model#

Functions#

load_scripted_model_from_uri(load_from_uri)

load_state_dict_from_uri(load_from_uri[, device])

save_scripted_model(model, save_to_path_uri)

save_state_dict(model, save_to_path_uri)

Module Contents#

gigl.src.common.utils.model.load_scripted_model_from_uri(load_from_uri)[source]#
Parameters:

load_from_uri (gigl.common.Uri)

Return type:

torch.nn.Module

gigl.src.common.utils.model.load_state_dict_from_uri(load_from_uri, device=torch.device('cpu'))[source]#
Parameters:
Return type:

collections.OrderedDict[str, torch.Tensor]

gigl.src.common.utils.model.save_scripted_model(model, save_to_path_uri)[source]#
Parameters:
Return type:

None

gigl.src.common.utils.model.save_state_dict(model, save_to_path_uri)[source]#
Parameters:
Return type:

None