This commit is contained in:
Peter Zhokhov
2018-10-30 14:11:38 -07:00
parent 3e3e2b7998
commit ef2e7246c9

View File

@@ -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):