cunyizhang的博客

import os import torch src_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: ...