marcoancona/TorchPruner

About the bugs

Opened this issue · 3 comments

Hi,

Thank you for your great work, I just read your PhD thesis. And when I ran your code, I faced the bug in the code:
from torchpruner import (Pruner, ShapleyAttributionMetric), can't import Pruner, can you have a look?

Thank you so much.

Best.

hi, have u solved this issue? I got the bug same with u

Oh, sorry, I didn't solve the bug.... :(

change into this:
from torchpruner.attributions import ShapleyAttributionMetric
from torchpruner.pruner import Pruner