Code bug? In mpi4pytorch.py

tomgoldstein / loss-landscape

Code for visualizing the loss landscape of neural nets

MIT License

2.72k stars 388 forks source link

Code bug? In mpi4pytorch.py #2

Closed jihaonew closed 5 years ago

jihaonew commented 5 years ago

mpi4pytorch.py The function's name is allreduce_min, but use MPI.MAX .

ljk628 commented 5 years ago

Good catch! Thanks a lot! Just fixed it via aa6bac7b70fe1787f445b5e8bc10b7f8e43b7b99. This function is deprecated and not used in plotting the surfaces.