diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-31 10:51:38 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-31 10:51:38 +0800 |
| commit | 07de5a9797a4f53779fe43c14ee51bdc4e248463 (patch) | |
| tree | cf0c7f57657766bae005c8182e07c7c291482f5e /models/layers.py | |
| parent | 2fddfd8f99f86f389117541421e457272f216d0b (diff) | |
Bug Fixes in HPM and PartNet
1. Register list of torch.nn.Module to the network using torch.nn.ModuleList
2. Fix operation error in squeeze list of tensor
3. Replace squeeze with view in HP in case batch size is 1
Diffstat (limited to 'models/layers.py')
0 files changed, 0 insertions, 0 deletions