(self)
| 580 | return mu + std * torch.randn_like(std) |
| 581 | |
| 582 | def clear_cache(self): |
| 583 | self._conv_num = count_conv3d(self.decoder) |
| 584 | self._conv_idx = [0] |
| 585 | self._feat_map = [None] * self._conv_num |
| 586 | #cache encode |
| 587 | self._enc_conv_num = count_conv3d(self.encoder) |
| 588 | self._enc_conv_idx = [0] |
| 589 | self._enc_feat_map = [None] * self._enc_conv_num |
| 590 | |
| 591 | |
| 592 | def _video_vae(pretrained_path=None, z_dim=None, device='cpu', **kwargs): |
no test coverage detected