Module — PyTorch 1.10.1 documentation
pytorch.org › docs › stableload_state_dict (state_dict, strict = True) [source] ¶ Copies parameters and buffers from state_dict into this module and its descendants. If strict is True, then the keys of state_dict must exactly match the keys returned by this module’s state_dict() function. Parameters. state_dict – a dict containing parameters and persistent buffers.