[question] replace inception with resnet18

[jS5t3r]

Can I just replace the inception with a pre-trained resnet18?

Would it run on the demo data seamlessly?

[WindVChen]

Hi @jS5t3r ,

Yes, you can seamlessly replace the Inception model with a pre-trained ResNet18. An easier way to achieve this is by adding your desired model to the model_selection function and the models_transfer_name list. Simply expand the script in other_attacks.py as follows:

def model_selection(name):
    ...
    elif name == "resnet_18":
        model = models.resnet18(pretrained=True)
    ...

def model_transfer(...):
    ...
        models_transfer_name = ["resnet_18", ...]
    ...

[jS5t3r]

Thanks for the fast response :)