diff --git a/stable_learning_control/algos/pytorch/lac/lac.py b/stable_learning_control/algos/pytorch/lac/lac.py index 8d784d60..af3f05f6 100644 --- a/stable_learning_control/algos/pytorch/lac/lac.py +++ b/stable_learning_control/algos/pytorch/lac/lac.py @@ -887,10 +887,10 @@ def lac( lr_alpha_final=1e-10, lr_labda_final=1e-10, lr_decay_type=DEFAULT_DECAY_TYPE, - lr_a_decay_type=DEFAULT_DECAY_TYPE, - lr_c_decay_type=DEFAULT_DECAY_TYPE, - lr_alpha_decay_type=DEFAULT_DECAY_TYPE, - lr_labda_decay_type=DEFAULT_DECAY_TYPE, + lr_a_decay_type=None, + lr_c_decay_type=None, + lr_alpha_decay_type=None, + lr_labda_decay_type=None, lr_decay_ref=DEFAULT_DECAY_REFERENCE, batch_size=256, replay_size=int(1e6), diff --git a/stable_learning_control/algos/pytorch/sac/sac.py b/stable_learning_control/algos/pytorch/sac/sac.py index 97b7aa29..aaa15fc4 100644 --- a/stable_learning_control/algos/pytorch/sac/sac.py +++ b/stable_learning_control/algos/pytorch/sac/sac.py @@ -758,9 +758,9 @@ def sac( lr_c_final=1e-10, lr_alpha_final=1e-10, lr_decay_type=DEFAULT_DECAY_TYPE, - lr_a_decay_type=DEFAULT_DECAY_TYPE, - lr_c_decay_type=DEFAULT_DECAY_TYPE, - lr_alpha_decay_type=DEFAULT_DECAY_TYPE, + lr_a_decay_type=None, + lr_c_decay_type=None, + lr_alpha_decay_type=None, lr_decay_ref=DEFAULT_DECAY_REFERENCE, batch_size=256, replay_size=int(1e6),