thomasverelst / dynconv

Code for Dynamic Convolutions: Exploiting Spatial Sparsity for Faster Inference (CVPR2020)
https://arxiv.org/abs/1912.03203
126 stars 14 forks source link

Ponder_Cost_Plotting #6

Open rida-rida opened 3 years ago

rida-rida commented 3 years ago

File "main_cifar.py", line 224, in validate viz.plot_ponder_cost(meta['masks']) File "/Data2/xyz/dynconv-master/classification/utils/viz.py", line 26, in plot_ponder_cost ponder_cost = ponder_cost_map(masks) File "/Data2/xyz/dynconv-master/classification/dynconv/utils.py", line 23, in ponder_cost_map return out.squeeze(0).cpu().numpy() AttributeError: 'NoneType' object has no attribute 'squeeze'

Why this is showing while plotting the ponder_cost_map?