Skip to content

concat network

조회 수 18 추천 수 0 2020.08.27 01:19:46

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


WHRIA

2020.09.05 12: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)

WHRIA

2020.10.08 14:08:52
*.0.0.1

https://gist.github.com/andrewjong/6b02ff237533b3b2c554701fb53d5c4d

List of Articles
번호 제목 글쓴이 날짜 조회 수
1695 dkms for r8125 WHRIA 2020-11-12 5
1694 unattended upgrade WHRIA 2020-11-01 124
1693 pytorch pretrained WHRIA 2020-10-28 27
1692 steamlit WHRIA 2020-10-15 60
1691 sample size WHRIA 2020-10-13 25
1690 Transformer WHRIA 2020-10-09 67
1689 file lock WHRIA 2020-09-22 44
1688 onnx broswer WHRIA 2020-09-15 15
1687 fda 인증 WHRIA 2020-09-03 18
1686 ubuntu cuda nvidia-smi WHRIA 2020-08-29 500
» concat network [3] WHRIA 2020-08-27 18
1684 GPT2 [1] WHRIA 2020-08-03 29
1683 scopus [1] WHRIA 2020-08-02 12
1682 melafind WHRIA 2020-08-01 340
1681 nvidia dali [1] WHRIA 2020-08-01 13

Powered by Xpress Engine / Designed by Sketchbook

sketchbook5, 스케치북5

sketchbook5, 스케치북5

나눔글꼴 설치 안내


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

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

설치 취소