geowatch.utils.lightning_ext package

Subpackages

• geowatch.utils.lightning_ext.callbacks package
  • Submodules
    • geowatch.utils.lightning_ext.callbacks.auto_resumer module
      • AutoResumer
    • geowatch.utils.lightning_ext.callbacks.batch_plotter module
      • BatchPlotter
    • geowatch.utils.lightning_ext.callbacks.packager module
      • Packager
      • default_save_package()
    • geowatch.utils.lightning_ext.callbacks.state_logger module
      • StateLogger
    • geowatch.utils.lightning_ext.callbacks.telemetry module
      • LightningTelemetry
    • geowatch.utils.lightning_ext.callbacks.tensorboard_plotter module
      • TensorboardPlotter
    • geowatch.utils.lightning_ext.callbacks.text_logger module
      • TextLogger
  • Module contents
    • AutoResumer
      • AutoResumer.on_init_start()
      • AutoResumer.recent_checkpoints()
    • BatchPlotter
      • BatchPlotter.add_argparse_args()
      • BatchPlotter.draw_batch()
      • BatchPlotter.draw_if_ready()
      • BatchPlotter.on_test_batch_end()
      • BatchPlotter.on_test_batch_start()
      • BatchPlotter.on_train_batch_end()
      • BatchPlotter.on_train_batch_start()
      • BatchPlotter.on_validation_batch_end()
      • BatchPlotter.on_validation_batch_start()
      • BatchPlotter.setup()
    • Packager
      • Packager.add_argparse_args()
      • Packager.on_exception()
      • Packager.on_fit_end()
      • Packager.on_fit_start()
      • Packager.setup()
    • StateLogger
      • StateLogger.on_exception()
      • StateLogger.on_fit_end()
      • StateLogger.on_fit_start()
      • StateLogger.on_sanity_check_end()
      • StateLogger.on_sanity_check_start()
      • StateLogger.on_save_checkpoint()
      • StateLogger.setup()
      • StateLogger.teardown()
    • TensorboardPlotter
      • TensorboardPlotter.on_test_epoch_end()
      • TensorboardPlotter.on_train_epoch_end()
      • TensorboardPlotter.on_validation_epoch_end()
    • TextLogger
      • TextLogger.load_state_dict()
      • TextLogger.on_exception()
      • TextLogger.on_fit_end()
      • TextLogger.on_fit_start()
      • TextLogger.on_sanity_check_end()
      • TextLogger.on_sanity_check_start()
      • TextLogger.on_train_end()
      • TextLogger.on_train_epoch_end()
      • TextLogger.on_train_epoch_start()
      • TextLogger.on_train_start()
      • TextLogger.on_validation_epoch_end()
      • TextLogger.on_validation_epoch_start()
      • TextLogger.setup()
      • TextLogger.state_dict()
      • TextLogger.teardown()
    • default_save_package()
    • LightningTelemetry
      • LightningTelemetry.add_argparse_args()
      • LightningTelemetry.on_exception()
      • LightningTelemetry.on_fit_end()
      • LightningTelemetry.on_fit_start()
      • LightningTelemetry.on_train_epoch_end()
      • LightningTelemetry.setup()

Submodules

• geowatch.utils.lightning_ext.argparse_ext module
  • get_init_arguments_and_types()
  • str_to_bool_or_str()
  • str_to_bool_or_int()
  • str_to_bool()
  • parse_docstring_args()
  • add_arginfos_to_parser()
  • add_argparse_args()
• geowatch.utils.lightning_ext.demo module
  • LightningToyNet2d
    • LightningToyNet2d.forward()
    • LightningToyNet2d.get_cfgstr()
    • LightningToyNet2d.forward_step()
    • LightningToyNet2d.validation_step()
    • LightningToyNet2d.training_step()
    • LightningToyNet2d.train_dataloader()
    • LightningToyNet2d.val_dataloader()
    • LightningToyNet2d.configure_optimizers()
  • demo_trainer()
• geowatch.utils.lightning_ext.lightning_cli_ext module
  • LightningCLI_Extension
    • LightningCLI_Extension.init_parser()
    • LightningCLI_Extension.parse_arguments()
• geowatch.utils.lightning_ext.monkeypatches module
• geowatch.utils.lightning_ext.old_parser_devices module
  • parse_gpu_ids()
• geowatch.utils.lightning_ext.util_device module
  • coerce_devices()
  • coerce_accelerator_devices()
• geowatch.utils.lightning_ext.util_globals module
• geowatch.utils.lightning_ext.util_model module
  • model_hparams()

Module contents

mkinit geowatch.utils.lightning_ext –noattr -w