sarvani30

Error while trying to prune Lenet-5 architecture with MNIST dataset

5

DG.build_dependency(model, example_inputs=th.randn(1,1,28,28)) File "/usr/local/lib/python3.6/dist-packages/torch_pruning/dependency.py", line 309, in build_dependency self.update_index() File "/usr/local/lib/python3.6/dist-packages/torch_pruning/dependency.py", line 315, in update_index self._set_fc_index_transform( node ) File "/usr/local/lib/python3.6/dist-packages/torch_pruning/dependency.py", line 437, in _set_fc_index_transform feature_channels = _get_in_node_out_channels(fc_node.inputs[0]) **IndexError: list index...