def load_checkpoint_in_model( model: nn.Module, checkpoint: Union[str, os.PathLike], device_map: Optional[Dict[str, Union[int, str, torch.device]]] = None, offload_folder: Optional[Union[str, os.PathLike]] = None, dtype: Optional[Union[str, torch.dtype]] = None, offload_state_dict...
Check whether the parameters in checkpoint are same as the load file. 👍 1 17Skye17 reacted with thumbs up emoji ParthikB commented Jul 15, 2019 Solved. add tf.reset_default_graph() before building the net and model. Reference: http://stackoverflow.com/questions/33765336/remove-nodes-...