Returns: bool: whether hvd is currently running under MPI
(self)
| 436 | self.BROADCAST_EVERY_EPOCH = True |
| 437 | |
| 438 | def mpi_enabled(self): |
| 439 | """ |
| 440 | Returns: |
| 441 | bool: whether hvd is currently running under MPI |
| 442 | """ |
| 443 | try: |
| 444 | return self.hvd.mpi_enabled() |
| 445 | except AttributeError: |
| 446 | return False |
| 447 | |
| 448 | def allreduce(self, grads): |
| 449 | if self.hvd.size() == 1: |
nothing calls this directly
no test coverage detected