2016-06-06 4 views
0

나는이 paper에서 설명한 것처럼 Caffe에서 연결 제거를 구현하려고합니다. 기본적으로 0에 가까운 모든 가중치는 순방향 및 역방향 전파에 간섭을 일으키지 않아야합니다. 저자는 "Caffe는 각 가중치 텐서에 대해 네트워크 작업 중 정리 된 매개 변수를 무시한 마스크를 추가하도록 수정되었습니다." 마스크가 들어있는 각 레이어에 여분의 blob을 추가했습니다. 이제 conv_layer.cpp와 inner_product_layer.cpp를 변경하여 여분의 blob을로드하고 제거 된 매개 변수를 무시하려고했습니다. 그러나이 레이어에 작성된 코드를 이해하지 못하고 이에 대한 문서를 찾을 수 없습니다. 아무도 나에게이 파일의 소스 코드를 설명 할 수 있습니까?Caffe Pruning Connections

답변