diff --git a/baselines/common/mpi_adam.py b/baselines/common/mpi_adam.py index fd10bd5..9168e5b 100644 --- a/baselines/common/mpi_adam.py +++ b/baselines/common/mpi_adam.py @@ -5,7 +5,7 @@ try: from mpi4py import MPI except ImportError: MPI = None - + class MpiAdam(object): def __init__(self, var_list, *, beta1=0.9, beta2=0.999, epsilon=1e-08, scale_grad_by_procs=True, comm=None):