diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-19 21:03:42 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2021-02-19 21:03:42 +0800 |
| commit | e33c22e556ed64e1c1fdb011d78a124d1489ad15 (patch) | |
| tree | 4f11c8fd2a9653e71d92b210754a9db2eb4ef0be /models/rgb_part_net.py | |
| parent | 4572598bca01d29b039f7d5291b08b53c1419528 (diff) | |
| parent | a31eb135c8cc3a8737fabca54fe3d5791f293753 (diff) | |
Merge branch 'data_parallel' into data_parallel_py3.8
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 538ca53..2af990e 100644 --- a/models/rgb_part_net.py +++ b/models/rgb_part_net.py @@ -86,7 +86,7 @@ class RGBPartNet(nn.Module): pn_ba_trip = self.pn_ba_trip( x[self.hpm_num_parts:], y[self.hpm_num_parts:] ) - losses = (*losses, hpm_ba_trip, pn_ba_trip) + losses = torch.stack((*losses, hpm_ba_trip, pn_ba_trip)) return losses, images else: return x.unsqueeze(1).view(-1) |