diff options
| author | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-30 22:16:22 +0800 |
|---|---|---|
| committer | Jordan Gong <jordan.gong@protonmail.com> | 2020-12-30 22:16:22 +0800 |
| commit | 2fddfd8f99f86f389117541421e457272f216d0b (patch) | |
| tree | b44922c780d251de2d7b4b5560e186c11b27c641 /models/hpm.py | |
| parent | 078657c1f42f62e5d3834d8a9f2c0226daae7320 (diff) | |
Correct and refine PartNet
1. Let FocalConv block capable of processing frames in all batches
2. Correct input dims of TFA and output dims of HP
3. Change torch.unsqueeze and torch.cat to torch.stack
Diffstat (limited to 'models/hpm.py')
0 files changed, 0 insertions, 0 deletions