Remove accel from config

NKI-AI · Feb 1, 2024 · eeac10a · eeac10a
1 parent ef6f359
commit eeac10a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/direct/nn/adaptive/config.py b/direct/nn/adaptive/config.py
@@ -14,8 +14,6 @@
 
 @dataclass
 class PolicyConfig(ModelConfig):
-    acceleration: float = MISSING
-    center_fraction: float = MISSING
     sampling_dimension: PolicySamplingDimension = MISSING
     st_slope: float = 10
     st_clamp: bool = False
@@ -42,11 +40,13 @@ class Parameterized3dPolicyConfig(ParameterizedPolicyConfig):
 @dataclass
 class ParameterizedDynamic2dPolicyConfig(ParameterizedPolicyConfig):
     num_time_steps: int = MISSING
+    non_uniform: bool = False
 
 
 @dataclass
 class ParameterizedMultislice2dPolicyConfig(ParameterizedPolicyConfig):
     num_slices: int = MISSING
+    non_uniform: bool = False
 
 
 @dataclass