[Question] Using a different model?

[ZhuoyueLyu]

Hi @haekyu, great work! Just curious, if we want to use models other than Inception v1, do we just need to update the LAYER_BLK_SIZES and LAYER_SIZES in the model.py? If not, do you have suggestions on what we need to update? Thanks!

[haekyu]

Hi Zhuoyue, thanks for your interest in this work. Yes you might want to update the variables in model.py. We haven't checked whether other models work thus it might need more modifications in other codes. For example, if you want to use the visualization, you might need to update the data (e.g., data/top-neurons/top_neurons-ambulance-street_sign-fgsm.json).