Ponder_Cost_Plotting

[rida-rida]

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?