| 17 | ## load config |
| 18 | @dataclass |
| 19 | class Args: |
| 20 | # exp |
| 21 | seed: int = 0 |
| 22 | disable_recommended_params: bool = False |
| 23 | not_render: bool = False |
| 24 | # env |
| 25 | env_name: str = ( |
| 26 | "ant" # "humanoidstandup", "ant", "halfcheetah", "hopper", "walker2d", "car2d" |
| 27 | ) |
| 28 | # diffusion |
| 29 | Nsample: int = 2048 # number of samples |
| 30 | Hsample: int = 50 # horizon |
| 31 | Ndiffuse: int = 100 # number of diffusion steps |
| 32 | temp_sample: float = 0.1 # temperature for sampling |
| 33 | beta0: float = 1e-4 # initial beta |
| 34 | betaT: float = 1e-2 # final beta |
| 35 | enable_demo: bool = False |
| 36 | |
| 37 | |
| 38 | def run_diffusion(args: Args): |
nothing calls this directly
no outgoing calls
no test coverage detected