fl4health.checkpointing package

Submodules

• fl4health.checkpointing.checkpointer module
  • BestLossTorchModuleCheckpointer
    • BestLossTorchModuleCheckpointer.__init__()
    • BestLossTorchModuleCheckpointer.maybe_checkpoint()
  • FunctionTorchModuleCheckpointer
    • FunctionTorchModuleCheckpointer.__init__()
    • FunctionTorchModuleCheckpointer.maybe_checkpoint()
  • LatestTorchModuleCheckpointer
    • LatestTorchModuleCheckpointer.__init__()
    • LatestTorchModuleCheckpointer.maybe_checkpoint()
  • PerRoundStateCheckpointer
    • PerRoundStateCheckpointer.__init__()
    • PerRoundStateCheckpointer.checkpoint_exists()
    • PerRoundStateCheckpointer.load_checkpoint()
    • PerRoundStateCheckpointer.save_checkpoint()
  • TorchModuleCheckpointer
    • TorchModuleCheckpointer.__init__()
    • TorchModuleCheckpointer.load_checkpoint()
    • TorchModuleCheckpointer.maybe_checkpoint()
• fl4health.checkpointing.client_module module
  • CheckpointMode
    • CheckpointMode.POST_AGGREGATION
    • CheckpointMode.PRE_AGGREGATION
  • ClientCheckpointAndStateModule
    • ClientCheckpointAndStateModule.__init__()
    • ClientCheckpointAndStateModule.maybe_checkpoint()
    • ClientCheckpointAndStateModule.maybe_load_state()
    • ClientCheckpointAndStateModule.save_state()
• fl4health.checkpointing.opacus_checkpointer module
  • BestLossOpacusCheckpointer
    • BestLossOpacusCheckpointer.__init__()
    • BestLossOpacusCheckpointer.maybe_checkpoint()
  • LatestOpacusCheckpointer
    • LatestOpacusCheckpointer.__init__()
    • LatestOpacusCheckpointer.maybe_checkpoint()
  • OpacusCheckpointer
    • OpacusCheckpointer.load_best_checkpoint_into_model()
    • OpacusCheckpointer.load_checkpoint()
    • OpacusCheckpointer.maybe_checkpoint()
• fl4health.checkpointing.server_module module
  • AdaptiveConstraintServerCheckpointAndStateModule
    • AdaptiveConstraintServerCheckpointAndStateModule.__init__()
  • BaseServerCheckpointAndStateModule
    • BaseServerCheckpointAndStateModule.__init__()
    • BaseServerCheckpointAndStateModule.maybe_checkpoint()
    • BaseServerCheckpointAndStateModule.maybe_load_state()
    • BaseServerCheckpointAndStateModule.save_state()
  • ClippingBitServerCheckpointAndStateModule
    • ClippingBitServerCheckpointAndStateModule.__init__()
  • DpScaffoldServerCheckpointAndStateModule
    • DpScaffoldServerCheckpointAndStateModule.__init__()
  • LayerNamesServerCheckpointAndStateModule
    • LayerNamesServerCheckpointAndStateModule.__init__()
  • NnUnetServerCheckpointAndStateModule
    • NnUnetServerCheckpointAndStateModule.__init__()
  • OpacusServerCheckpointAndStateModule
    • OpacusServerCheckpointAndStateModule.__init__()
  • PackingServerCheckpointAndAndStateModule
    • PackingServerCheckpointAndAndStateModule.__init__()
  • ScaffoldServerCheckpointAndStateModule
    • ScaffoldServerCheckpointAndStateModule.__init__()
  • SparseCooServerCheckpointAndStateModule
    • SparseCooServerCheckpointAndStateModule.__init__()