张文龙的博客

资料

def load_network(self, load_path, network, strict=True): if isinstance(network, nn.DataParallel): network = network.module model_dict = torch.load(load_path) filtered = {k: v for ...