Create config.py

zzmtsvv · Sep 4, 2023 · 1c44000 · 1c44000
1 parent 0eefa46
commit 1c44000
Showing 1 changed file with 37 additions and 0 deletions.
diff --git a/prdc/config.py b/prdc/config.py
@@ -0,0 +1,37 @@
+from dataclasses import dataclass
+import torch
+
+
+@dataclass
+class prdc_config:
+    # Experiment
+    device: str = "cuda" if torch.cuda.is_available() else "cpu"
+    dataset_name: str = "halfcheetah-medium-v2"
+    seed: int = 42
+    max_timesteps: int = int(1e6)
+
+    max_action : float = 1.0
+
+    action_dim: int = 6
+    state_dim: int = 17
+
+    buffer_size: int = 1_000_000
+    actor_lr: float = 3e-4
+    critic_lr: float = 3e-4
+    hidden_dim: int = 256
+    batch_size: int = 256
+    discount: float = 0.99
+    tau: float = 0.005
+    policy_noise: float = 0.2
+    noise_clip: float = 0.5
+    policy_freq: int = 2
+
+    alpha: float = 40.0
+    beta: float = 2
+    k: int = 1
+    critic_ln: bool = True
+    normalize: bool = True
+
+    project: str = "PRDC"
+    group: str = dataset_name
+    name: str = dataset_name + "_" + str(seed)