L2 계층 이후에 많은 계층이있는 네트워크에서 L2 (중간 계층) 표현에 손실 함수를 적용해야합니다. 다음과 같이 nngraph에서 네트워크 출력에 액세스하는 방법을 알고 있습니다.
input = nn.Identity()()
net = nn.Sequential()
net:add(nn.Linear(100, 20)):add(nn.ReLU(true)) -- L1
net:add(nn.Linear(20, 10)):add(ReLU(true)) -- L2
net:add(nn.Linear(10, 2)) -- L3
output = net(input)
gmod = nn.gModule({input}, {output})
그러나 두 번째 레이어의 결과에 액세스하고 손실 함수 (기준)를 적용하고 깔끔하게 역 전파를 수행하는 방법을 모르겠습니다. 누구든지 이것에 대해 나에게 도움을 줄 수 있습니까?
레이어를 별도의 출력으로 지정하면 언제든지 액세스 할 수 있습니다.
input = nn.Identity()()
L1 = nn.ReLU(true)(nn.Linear(100, 20)(input))
L2 = nn.ReLU(true)(nn.Linear(20, 10)(L1))
L3 = nn.Linear(10, 2)(L2)
gmod = nn.gModule({input}, {L3, L2})
이 기사는 인터넷에서 수집됩니다. 재 인쇄 할 때 출처를 알려주십시오.
침해가 발생한 경우 연락 주시기 바랍니다[email protected] 삭제
몇 마디 만하겠습니다