부캠 TIL 0914 - CNN visualization

numel

파라미터 개수구할 수 있다.

param.numel()

get layer

pre trained model의 레이어를 지정하는 방법은

conv1_filters_data = model.backbone.conv1.weight

와 같이 해주면 된다.

backbone -> conv1 순서대로 들어있기 때문에 model을 찍어보고 구조를 본다음 지정하면된다.

그래서 이러한 레이어들이 attribute로 들어가있는지 찍어봤더니

요런식으로 찍힌다.

ordered dict로 되어있어서 저렇게 계층적으로 타고 들어가면서 인덱싱 해줄 수 있는 것 같다.