diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-22 20:51:08 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-22 20:51:08 +0800 |
| commit | 231a14649a8c88463610c69c770b0397992b1210 (patch) | |
| tree | c860dddac3a8a7e18ed48dff8186c0300dc4cc4d /models/rgb_part_net.py | |
| parent | e77d74684a5b77b4b5ce1edac67518fea12bd5c1 (diff) | |
| parent | 4a4506fde7867df4d4d446c8992621fb0da03939 (diff) | |
Merge branch 'python3.8' into python3.7
# Conflicts:
# utils/configuration.py
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 fba3485..fcd8fbc 100644 --- a/models/rgb_part_net.py +++ b/models/rgb_part_net.py @@ -59,7 +59,7 @@ class RGBPartNet(nn.Module): if self.training: return x_c, x_p, ae_losses, images else: - return torch.cat((x_c.view(-1), x_p.view(-1))) + return x_c, x_p def _disentangle(self, x_c1_t2, x_c2_t2=None): n, t, c, h, w = x_c1_t2.size() |