Add an option Uniform.round to round the initial numbers.

mart/attack/initializer/base.py

@@ -40,13 +40,16 @@ def initialize_(self, parameter: torch.Tensor) -> None:
 
 
 class Uniform(Initializer):
-    def __init__(self, min: int | float, max: int | float):
+    def __init__(self, min: int | float, max: int | float, round: bool = False):
         self.min = min
         self.max = max
+        self.round = round
 
     @torch.no_grad()
     def initialize_(self, parameter: torch.Tensor) -> None:
         torch.nn.init.uniform_(parameter, self.min, self.max)
+        if self.round:
+            parameter.round_()
 
 
 class UniformLp(Initializer):

mart/configs/attack/composer/perturber/initializer/uniform.yaml

@@ -1,3 +1,4 @@
 _target_: mart.attack.initializer.Uniform
 min: ???
 max: ???
+round: false