fl4health.checkpointing package

Submodules

• fl4health.checkpointing.checkpointer module
  • BestLossTorchModuleCheckpointer
    • BestLossTorchModuleCheckpointer.__init__()
    • BestLossTorchModuleCheckpointer.maybe_checkpoint()
  • BestMetricTorchModuleCheckpointer
    • BestMetricTorchModuleCheckpointer.__init__()
  • FunctionTorchModuleCheckpointer
    • FunctionTorchModuleCheckpointer.__init__()
    • FunctionTorchModuleCheckpointer.maybe_checkpoint()
  • LatestTorchModuleCheckpointer
    • LatestTorchModuleCheckpointer.__init__()
    • LatestTorchModuleCheckpointer.maybe_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__()
• fl4health.checkpointing.state_checkpointer module
  • ClientStateCheckpointer
    • ClientStateCheckpointer.__init__()
    • ClientStateCheckpointer.get_attribute()
    • ClientStateCheckpointer.maybe_load_client_state()
    • ClientStateCheckpointer.maybe_set_default_checkpoint_name()
    • ClientStateCheckpointer.save_client_state()
    • ClientStateCheckpointer.set_attribute()
  • NnUnetServerStateCheckpointer
    • NnUnetServerStateCheckpointer.__init__()
  • ServerStateCheckpointer
    • ServerStateCheckpointer.__init__()
    • ServerStateCheckpointer.get_attribute()
    • ServerStateCheckpointer.maybe_load_server_state()
    • ServerStateCheckpointer.maybe_set_default_checkpoint_name()
    • ServerStateCheckpointer.save_server_state()
    • ServerStateCheckpointer.set_attribute()
  • StateCheckpointer
    • StateCheckpointer.__init__()
    • StateCheckpointer.add_to_snapshot_attr()
    • StateCheckpointer.checkpoint_exists()
    • StateCheckpointer.delete_from_snapshot_attr()
    • StateCheckpointer.get_attribute()
    • StateCheckpointer.load_checkpoint()
    • StateCheckpointer.load_state()
    • StateCheckpointer.save_checkpoint()
    • StateCheckpointer.save_state()
    • StateCheckpointer.set_attribute()
    • StateCheckpointer.set_checkpoint_path()