We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 47ba1e1 commit 93e5fbcCopy full SHA for 93e5fbc
1 file changed
src/acrl/agent.py
@@ -20,7 +20,6 @@
20
)
21
from acrl.utils.state import EnvironmentState
22
from acrl.utils.warmup import SpeedRewardWarmUp
23
-
24
import numpy as np
25
import torch
26
import wandb
@@ -200,7 +199,6 @@ def setup(self):
200
199
self._setup_checkpointer()
201
self._setup_reward_warmup()
202
203
204
def _unpack_config(self):
205
self._n_step_buffer_states = self.cfg["sac"]["n_steps"]
206
self._start_steps = self.cfg["training"]["start_steps"]
@@ -260,4 +258,3 @@ def _setup_reward_warmup(self):
260
258
config = self.cfg["training"]["speed_reward_warmup"]
261
259
self._reward_warmup = SpeedRewardWarmUp(config)
262
self._maximum_speed = config["max_speed_reward"]["final"]
263
0 commit comments