PyTorch layer별 값 비교, 출력

Serendipity·2023년 9월 17일
0

2023 LeSN

목록 보기
23/52

Layer 별로 tensor값 출력하기

https://discuss.pytorch.org/t/how-can-i-extract-intermediate-layer-output-from-loaded-cnn-model/77301/3

모델

모델 로드

https://discuss.pytorch.org/t/how-can-i-extract-intermediate-layer-output-from-loaded-cnn-model/77301/28

referring to the weights by neurons
model.layer.weight, model.layer.bias -> add them to constraints operations.

profile
I'm an graduate student majoring in Computer Engineering at Inha University. I'm interested in Machine learning developing frameworks, Formal verification, and Concurrency.

0개의 댓글