Fixed: Upgrade changing interfaces
This commit is contained in:
parent
f3e03916c8
commit
844cdb1319
@ -134,31 +134,31 @@ class SAC(OffPolicyAlgorithm):
|
||||
_init_setup_model: bool = True,
|
||||
):
|
||||
super().__init__(
|
||||
policy,
|
||||
env,
|
||||
None, # PolicyBase
|
||||
learning_rate,
|
||||
buffer_size,
|
||||
learning_starts,
|
||||
batch_size,
|
||||
tau,
|
||||
gamma,
|
||||
train_freq,
|
||||
gradient_steps,
|
||||
action_noise,
|
||||
policy=policy,
|
||||
env=env,
|
||||
learning_rate=learning_rate,
|
||||
buffer_size=buffer_size,
|
||||
learning_starts=learning_starts,
|
||||
batch_size=batch_size,
|
||||
tau=tau,
|
||||
gamma=gamma,
|
||||
train_freq=train_freq,
|
||||
gradient_steps=gradient_steps,
|
||||
action_noise=action_noise,
|
||||
replay_buffer_class=replay_buffer_class,
|
||||
replay_buffer_kwargs=replay_buffer_kwargs,
|
||||
policy_kwargs=policy_kwargs,
|
||||
tensorboard_log=tensorboard_log,
|
||||
verbose=verbose,
|
||||
device=device,
|
||||
support_multi_env=False,
|
||||
monitor_wrapper=True,
|
||||
seed=seed,
|
||||
use_sde=use_sde,
|
||||
sde_sample_freq=sde_sample_freq,
|
||||
use_sde_at_warmup=use_sde_at_warmup,
|
||||
optimize_memory_usage=optimize_memory_usage,
|
||||
sde_support=True,
|
||||
supported_action_spaces=(gym.spaces.Box),
|
||||
support_multi_env=True,
|
||||
)
|
||||
|
||||
self.target_entropy = target_entropy
|
||||
|
Loading…
Reference in New Issue
Block a user