From 8f9bdd3ba1ac09441ac74a75b222a71e7c112ce5 Mon Sep 17 00:00:00 2001 From: Andrew White Date: Mon, 27 Jul 2026 07:32:52 -0500 Subject: [PATCH] fix: resnet_bpda baseline not set to eval mode --- eval_sde_adv_bpda.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/eval_sde_adv_bpda.py b/eval_sde_adv_bpda.py index 6414059..2e34c84 100644 --- a/eval_sde_adv_bpda.py +++ b/eval_sde_adv_bpda.py @@ -128,7 +128,7 @@ def eval_bpda(args, config, model, x_val, y_val, adv_batch_size, log_dir): # ------------------ apply the attack to resnet ------------------ print(f'apply the bpda attack to resnet...') - resnet_bpda = ResNet_Adv_Model(args, config) + resnet_bpda = ResNet_Adv_Model(args, config).eval() if ngpus > 1: resnet_bpda = torch.nn.DataParallel(resnet_bpda)