diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-25 15:46:31 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-03-25 15:46:31 +0800 |
| commit | 8dd0c42dc341445987c6a907df909155a8b9abd2 (patch) | |
| tree | c8b2ceedbb2513662b4cf151121ddf3ca2deb187 /models/rgb_part_net.py | |
| parent | 478ecf50f3dd9e2c2f8ca62e6ecd4a65b2cc7c3a (diff) | |
| parent | 104c6fbf0686828ed299b2a8bda1806a9b45f440 (diff) | |
Merge branch 'data_parallel' into data_parallel_py3.8data_parallel_py3.8
# Conflicts:
# models/model.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 81f198e..e8e320d 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.transpose(0, 1), x_p.transpose(0, 1), images, f_loss else: - return torch.cat((x_c, x_p)).unsqueeze(1).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() |