Advices on customize SUMMIT to other models and datasets, eg. PyTorch models?

Hi @fredhohman , SUMMIT is a really cool tool for investigating globally the feature maps. It would be nice to extend it to other models and datasets.

I wanted to use your notebooks and my pytorch model to generate data for the javascript pipeline but found that they depend on older versions of Tensorflow and Lucid.

Could you provide the version dependencies for the notebooks? Could you also give some advices if one wants to extend it to pytorch models?

Thanks a lot!