[PyTorch] model.state.dict() 파라미터 확인하기

 저장한 state_dict 파일을 다음과 같이 불러와서 파라미터를 확인할 수 있다.

 

state_dict = torch.load(config['init_model'])
for param_tensor in state_dict:
    print(param_tensor, '\t', state_dict[param_tensor].size())