summaryrefslogtreecommitdiff
path: root/test/hpm.py
diff options
context:
space:
mode:
authorJordan Gong <jordan.gong@protonmail.com>2020-12-30 22:16:22 +0800
committerJordan Gong <jordan.gong@protonmail.com>2020-12-30 22:16:22 +0800
commit2fddfd8f99f86f389117541421e457272f216d0b (patch)
treeb44922c780d251de2d7b4b5560e186c11b27c641 /test/hpm.py
parent078657c1f42f62e5d3834d8a9f2c0226daae7320 (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 'test/hpm.py')
0 files changed, 0 insertions, 0 deletions