diff --git a/rl_games/algos_torch/a2c_continuous.py b/rl_games/algos_torch/a2c_continuous.py index b321e5af..2d46eaea 100644 --- a/rl_games/algos_torch/a2c_continuous.py +++ b/rl_games/algos_torch/a2c_continuous.py @@ -213,7 +213,6 @@ def calc_gradients(self, input_dict): self.aux_loss_dict[k] = v.detach() else: self.aux_loss_dict[k] = [v.detach()] - if self.multi_gpu: self.optimizer.zero_grad() else: diff --git a/rl_games/algos_torch/models.py b/rl_games/algos_torch/models.py index db2fcd1e..05bc4b4f 100644 --- a/rl_games/algos_torch/models.py +++ b/rl_games/algos_torch/models.py @@ -62,7 +62,6 @@ def denorm_value(self, value): def get_aux_loss(self): return None - class ModelA2C(BaseModel): def __init__(self, network): BaseModel.__init__(self, 'a2c')