Error while running the notebook dissect_classifier_experiment.ipynb

[Vikramank]

I encountered the following error while running dissect_classifier_experiment.ipynb

compute_acts() takes 1 positional argument but 2 were given

while executing the following cell
unit_images = iv.masked_images_for_topk( compute_acts, dataset, topk, k=5, num_workers=30, pin_memory=True, cachefile=resfile('top5images.npz'))

I checked out the helper functions. Could you please help me?

[choutianxius]

In the same code cell, just change
def compute_acts(image_batch):
to
def compute_acts(image_batch, *args):