Remove testing parameters

spencerwooo · Nov 20, 2024 · 52701fa · 52701fa
1 parent 503b338
commit 52701fa
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/torchattack/tgr.py b/torchattack/tgr.py
@@ -357,7 +357,5 @@ def mlp_tgr(
         TGR,
         model_name='vit_base_patch16_224',
         victim_model_names=['cait_s24_224', 'visformer_small'],
-        max_samples=24,
-        batch_size=4,
         from_timm=True,
     )
diff --git a/torchattack/vdc.py b/torchattack/vdc.py
@@ -711,7 +711,5 @@ def attn_add_vis(
         VDC,
         model_name='pit_b_224',
         victim_model_names=['cait_s24_224', 'visformer_small'],
-        max_samples=24,
-        batch_size=4,
         from_timm=True,
     )