DICTIONARY 썸네일형 리스트형 Pytorch how to use nn.ModuleDict with zip for iteration class TEST(nn.Module): def __init__(self): super().__init__() self.conv = nn.Conv2d(10, 10, 1) def forward(self, x): return self.conv(x) num_res = 2 BEVEncoder = nn.ModuleDict() UpSampler = nn.ModuleDict() for _ in range(num_res): IsSelfAttn = True if _ == 0 else False BEVEncoder[str(_)] = TEST() if (_ == 0): UpSampler[str(_)] = None else: UpSampler[str(_)] = TEST() for (_, enc), (_, up) in sort.. 더보기 For문을 이용한 Dictionary의 key, item 접근 # method 1 for key, value in batch.items(): if (exp_dim): batch[key] = batch[key].unsqueeze(0).cuda() else: batch[key] = batch[key].cuda() # method 2 for _, (key, value) in enumerate(batch.items()): if (exp_dim): batch[key] = batch[key].unsqueeze(0).cuda() else: batch[key] = batch[key].cuda() 더보기 이전 1 다음