cunyizhang的博客

import osimport torchsrc_weight = 'epoch_12.pth'model = torch.load(src_weight)weight = model['state_dict']filter_names = []layer_names = list(weight.keys())for layer_name in layer_names: ...