API documentation

API documentation

• Config
• Client
  • HeartbeatTimeouts
  • RankMonitorClient
    • RankMonitorClient.calculate_and_set_timeouts()
    • RankMonitorClient.init_workload_monitoring()
    • RankMonitorClient.load_state_dict()
    • RankMonitorClient.send_heartbeat()
    • RankMonitorClient.shutdown_workload_monitoring()
    • RankMonitorClient.state_dict()
  • RankMonitorClientError
• Server
  • RankMonitorServer
• Callback
  • FaultToleranceCallback
    • FaultToleranceCallback.on_exception()
    • FaultToleranceCallback.on_load_checkpoint()
    • FaultToleranceCallback.on_save_checkpoint()
    • FaultToleranceCallback.on_train_batch_end()
    • FaultToleranceCallback.on_train_end()
    • FaultToleranceCallback.on_train_start()
    • FaultToleranceCallback.on_validation_batch_end()
    • FaultToleranceCallback.on_validation_end()
    • FaultToleranceCallback.on_validation_start()
    • FaultToleranceCallback.setup()
    • FaultToleranceCallback.teardown()
  • SimulatedFaultParams
    • SimulatedFaultParams.fault_type
    • SimulatedFaultParams.base_delay
    • SimulatedFaultParams.rand_delay
    • SimulatedFaultParams.rank_to_fail