[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!