UOMOP

import torchimport torch.nn as nnfrom thop import profileclass Encoder(nn.Module): def __init__(self, latent_dim): super(Encoder, self).__init__() self.latent_dim = latent_dim # stride=2를 초반 레이어에 적용하고, kernel_size를 3으로 줄임 self.in1 = nn.Conv2d(3, 32, kernel_size=3, stride=2, padding=0) self.in2 = nn.Conv2d(3, 32, kernel_size=3, stride=2, padding=2) sel..