[Pytorch] insert new layer or block in the middle of pre-trained model
논문 구현을 하다보면 backbone model로 pre-trained model을 사용하는 경우가 많다. 방대한 양의 데이터를 처음부터 직접 학습시키는데 많은 시간이 소요되기 때문이다. 따라서 기존의 학습된 모델에 논문 저자의 아이디어를 적용시키는 경우가 많다. 모든 코드를 직접 작성해 모델의 architecture를 구성하고 그 위에 weight parameter를 입히는 방법도 있다. 하지만 block module과 같이 새로운 layer를 간단하게 추가하여 실험 환경을 빠르게 구축하는 방법도 있다. import torch from torch import nn # 학습된 모델 불러오기 qwer = torch.hub.load('facebookresearch/pytorchvideo', model = '..
