fix(eval-sde-adv): classifier not switched to eval mode (+1 more)

[andrewwhitecdw]

Small fixes in eval_sde_adv.py:

fix: classifier not switched to eval mode before AutoAttack

Fix: Replace:

    classifier = get_image_classifier(args.classifier_name).to(config.device)
    adversary_resnet = AutoAttack(classifier, norm=args.lp_norm, eps=args.adv_eps,

with:

    classifier = get_image_classifier(args.classifier_name).to(config.device).eval()
    adversary_resnet = AutoAttack(classifier, norm=args.lp_norm, eps=args.adv_eps,

fix: resnet baseline not in eval mode for stadv attack

Fix: Replace:

    resnet = get_image_classifier(args.classifier_name).to(config.device)

    start_time = time.time()
    init_acc = get_accuracy(resnet, x_val, y_val, bs=adv_batch_size)

with:

    resnet = get_image_classifier(args.classifier_name).to(config.device).eval()

    start_time = time.time()
    init_acc = get_accuracy(resnet, x_val, y_val, bs=adv_batch_size)

Files changed

• eval_sde_adv.py