change int to float parameters

surrogate.py

@@ -41,24 +41,24 @@ PARAM_BOUNDS = [
 
 
 class Parameterization(BaseModel):
-    N: int
+    N: float  # int
     alpha: float
-    d_model: int
-    dim_feedforward: int
+    d_model: float  # int
+    dim_feedforward: float  # int
     dropout: float
     emb_scaler: float
-    epochs_step: int
+    epochs_step: float  # int
     eps: float
     fudge: float
-    heads: int
-    k: int
+    heads: float  # int
+    k: float  # int
     lr: float
-    pe_resolution: int
-    ple_resolution: int
+    pe_resolution: float  # int
+    ple_resolution: float  # int
     pos_scaler: float
-    weight_decay: int
-    batch_size: int
-    out_hidden4: int
+    weight_decay: float  # int
+    batch_size: float  # int
+    out_hidden4: float  # int
     betas1: float
     betas2: float
     bias: bool