mldas.explore.loading.load_model

mldas.explore.loading. load_model ( fname , model ) [source]

Load saved model’s parameter dictionary to initialized model. The function will remove any .module string from parameter’s name.

Parameters
fname str

Path to saved model

model torch.nn.Module

Initialized network network architecture

Returns
model torch.nn.Module

Up-to-date neural network model