Hello, thanks for open-sourcing the great work!
I'm following the example on the main README for Mixed quantization + 2:4 pruning. I can run until python spdy.py rn18 imagenet 8 mixed --dp without any problems, but running the DP step throws an error:
Traceback (most recent call last):
File "spdy.py", line 240, in <module>
print(get_score(np.ones(len(layers))))
File "spdy.py", line 204, in get_score
solution = dp(costs)
File "spdy.py", line 182, in dp
solution.append(PD[layer][timing])
IndexError: index -10219 is out of bounds for axis 0 with size 10001
The only changes I made were changing the path to imagenet and commenting out lines 233-248 in database.py. Is this a version incompatibility issue? Any help is appreciated!
Hello, thanks for open-sourcing the great work!
I'm following the example on the main README for Mixed quantization + 2:4 pruning. I can run until
python spdy.py rn18 imagenet 8 mixed --dpwithout any problems, but running the DP step throws an error:The only changes I made were changing the path to imagenet and commenting out lines 233-248 in database.py. Is this a version incompatibility issue? Any help is appreciated!