github code 출처 : github/AntixK/PyTorch-VAE def forward(self, input: Tensor, **kwargs) -> Tensor: mu, log_var = self.encode(input) z = self.reparameterize(mu, log_var) return [self.decode(z), input, mu, log_var] 전체 beta-VAE의 구조. encode->reparam->decode. 여기서 encode의 output : mu(128 dim), log_var(128 dim) z : 128 dim(yaml file을 보면 됨.) encode의 input size : [N C H W]. model_params: name: 'BetaVAE' in..