fix data types from int to float

ain-soph · May 26, 2022 · fb68f5f · fb68f5f
1 parent 0fde5da
commit fb68f5f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/trojanvision/configs/defense/neuron_inspect.yml b/trojanvision/configs/defense/neuron_inspect.yml
@@ -1,6 +1,6 @@
 lambd_sp: 1.e-6
 lambd_sm: 1.e-6
-lambd_pe: 1
+lambd_pe: 1.
 thre: 1.e-5
 
 sample_ratio: 0.1
diff --git a/trojanvision/defenses/backdoor/model_inspection/neuron_inspect.py b/trojanvision/defenses/backdoor/model_inspection/neuron_inspect.py
@@ -26,8 +26,8 @@ def add_argument(cls, group: argparse._ArgumentGroup):
         group.add_argument('--sample_ratio', type=float, help='sample ratio from the full training data')
         return group
 
-    def __init__(self, lambd_sp: float = 1e-5, lambd_sm: float = 1e-5, lambd_pe: float = 1,
-                 thre: float = 0, sample_ratio: float = 0.1, **kwargs):
+    def __init__(self, lambd_sp: float = 1e-5, lambd_sm: float = 1e-5, lambd_pe: float = 1.,
+                 thre: float = 0., sample_ratio: float = 0.1, **kwargs):
         super().__init__(**kwargs)
         self.param_list['neuron_inspect'] = ['lambd_sp', 'lambd_sm', 'lambd_pe', 'thre', 'sample_ratio']