def handle_failure(self):
    self.shutdown()
    raise RuntimeError('Worker crashed during training. Training unsuccessful.')
def handle_failure(self):
    self.shutdown()
    raise RuntimeError('Worker crashed during training. Training unsuccessful.')