diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-14 21:07:28 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-14 21:07:28 +0800 |
| commit | da922be042d96338a3f207386e410b6746d046f5 (patch) | |
| tree | 5bf653641a3ed89680340718e20c66e3ee21c7a6 /models/rgb_part_net.py | |
| parent | 263b8001ca1b25a43d1c87f187423054e141925d (diff) | |
Bug fix when transforming and new config
Diffstat (limited to 'models/rgb_part_net.py')
| -rw-r--r-- | models/rgb_part_net.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/models/rgb_part_net.py b/models/rgb_part_net.py index c38a567..c136040 100644 --- a/models/rgb_part_net.py +++ b/models/rgb_part_net.py @@ -57,7 +57,7 @@ class RGBPartNet(nn.Module): if self.training: return x_c, x_p, ae_losses, images else: - return torch.cat((x_c, x_p)).unsqueeze(1).view(-1) + return torch.cat((x_c.view(-1), x_p.view(-1))) def _disentangle(self, x_c1_t2, x_c2_t2=None): n, t, c, h, w = x_c1_t2.size() |