str/config.py

from math import log
from dataclasses import dataclass
import torch


@dataclass
class str_config:
    # Experiment
    device: str = "cuda" if torch.cuda.is_available() else "cpu"
    dataset_name: str = "halfcheetah-medium-v2"
    seed: int = 42

    state_dim: int = 17
    action_dim: int = 6

    actor_lr: float = 3e-4
    critic_lr: float = 3e-4
    behavior_lr: float = 3e-4
    batch_size: int = 256
    buffer_size: int = 1000000
    discount: float = 0.99
    hidden_dim: int = 256
    max_action: float = 1.0
    max_timesteps: int = int(1e6)
    tau: float = 5e-3
    temperature: float = 0.5
    num_critics: int = 4

    advantage_min: float = -float("inf")
    advantage_max: float = log(100.0)

    project: str = "Offline STR"
    group: str = dataset_name
    name: str = dataset_name + "_" + str(seed)