Skip to content

concat network

조회 수 5 추천 수 0 2020.08.27 00:19:46

https://discuss.pytorch.org/t/concatenate-layer-output-with-additional-input-data/20462


WHRIA

2020.09.05 11:17:07
*.34.41.226

class MyModelA(nn.Module): def __init__(self): super(MyModelA, self).__init__() self.fc1 = nn.Linear(10, 2) def forward(self, x): x = self.fc1(x) return x class MyModelB(nn.Module): def __init__(self): super(MyModelB, self).__init__() self.fc1 = nn.Linear(20, 2) def forward(self, x): x = self.fc1(x) return x class MyEnsemble(nn.Module): def __init__(self, modelA, modelB): super(MyEnsemble, self).__init__() self.modelA = modelA self.modelB = modelB self.classifier = nn.Linear(4, 2) def forward(self, x1, x2): x1 = self.modelA(x1) x2 = self.modelB(x2) x = torch.cat((x1, x2), dim=1) x = self.classifier(F.relu(x)) return x # Create models and load state_dicts modelA = MyModelA() modelB = MyModelB() # Load state dicts modelA.load_state_dict(torch.load(PATH)) modelB.load_state_dict(torch.load(PATH)) model = MyEnsemble(modelA, modelB) x1, x2 = torch.randn(1, 10), torch.randn(1, 20) output = model(x1, x2)

List of Articles
번호 제목 글쓴이 날짜 조회 수
1689 file lock WHRIA 2020-09-22 3
1688 onnx broswer WHRIA 2020-09-15 3
1687 fda 인증 WHRIA 2020-09-03 7
1686 ubuntu cuda nvidia-smi WHRIA 2020-08-29 217
» concat network [2] WHRIA 2020-08-27 5
1684 GPT2 [1] WHRIA 2020-08-03 16
1683 scopus [1] WHRIA 2020-08-02 6
1682 melafind WHRIA 2020-08-01 11
1681 nvidia dali [1] WHRIA 2020-08-01 5
1680 pytorch optimize WHRIA 2020-08-01 5
1679 startup , 미국 WHRIA 2020-07-30 13
1678 consort , stard WHRIA 2020-07-30 2
1677 model split [1] WHRIA 2020-07-30 6
1676 FTC [1] WHRIA 2020-07-28 7
1675 암호화 WHRIA 2020-07-26 11

Powered by Xpress Engine / Designed by Sketchbook

sketchbook5, 스케치북5

sketchbook5, 스케치북5

나눔글꼴 설치 안내


이 PC에는 나눔글꼴이 설치되어 있지 않습니다.

이 사이트를 나눔글꼴로 보기 위해서는
나눔글꼴을 설치해야 합니다.

설치 취소