From be34d52327032e19a005ab45f219fafef82a95b1 Mon Sep 17 00:00:00 2001 From: Rahul Date: Sun, 5 Jun 2022 19:56:28 +0800 Subject: [PATCH] change the type of defense_remask_lr from int to float --- trojanvision/defenses/abstract.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/trojanvision/defenses/abstract.py b/trojanvision/defenses/abstract.py index c567a256..b3c87ad1 100644 --- a/trojanvision/defenses/abstract.py +++ b/trojanvision/defenses/abstract.py @@ -281,7 +281,7 @@ class ModelInspection(BackdoorDefense): Args: defense_remask_epoch (int): Defense watermark optimizing epochs. Defaults to ``10``. - defense_remask_lr (int): Defense watermark optimizing learning rate. + defense_remask_lr (float): Defense watermark optimizing learning rate. Defaults to ``0.1``. cost (float): Cost of mask norm loss. Defaults to ``1e-3``. @@ -299,7 +299,7 @@ def add_argument(cls, group: argparse._ArgumentGroup): group.add_argument('--defense_remask_epoch', type=int, help='defense watermark optimizing epochs ' '(default: 10)') - group.add_argument('--defense_remask_lr', type=int, + group.add_argument('--defense_remask_lr', type=float, help='defense watermark optimizing learning rate ' '(default: 0.1)') group.add_argument('--cost', type=float,