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
      • _torch_package_monkeypatch()
      • 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
      • _InstanceLogger
      • _strip_ansi()
  • Module contents
    • AutoResumer
      • AutoResumer.on_init_start()
      • AutoResumer.recent_checkpoints()
    • BatchPlotter
      • BatchPlotter.setup()
      • BatchPlotter.add_argparse_args()
      • BatchPlotter.draw_batch()
      • BatchPlotter.draw_if_ready()
      • BatchPlotter.on_train_batch_end()
      • BatchPlotter.on_validation_batch_end()
      • BatchPlotter.on_test_batch_end()
    • Packager
      • Packager.add_argparse_args()
      • Packager.setup()
      • Packager._after_initialization()
      • Packager.on_fit_start()
      • Packager.on_fit_end()
      • Packager.on_exception()
      • Packager._make_package_fpath()
      • Packager._save_package()
    • StateLogger
      • StateLogger.setup()
      • StateLogger.teardown()
      • StateLogger.on_fit_start()
      • StateLogger.on_fit_end()
      • StateLogger.on_save_checkpoint()
      • StateLogger.on_sanity_check_start()
      • StateLogger.on_sanity_check_end()
      • StateLogger.on_exception()
    • TensorboardPlotter
      • TensorboardPlotter._on_epoch_end()
      • TensorboardPlotter.on_train_epoch_end()
      • TensorboardPlotter.on_validation_epoch_end()
      • TensorboardPlotter.on_test_epoch_end()
    • TextLogger
      • TextLogger.setup()
      • TextLogger.teardown()
      • TextLogger.on_fit_start()
      • TextLogger.on_fit_end()
      • TextLogger.state_dict()
      • TextLogger.load_state_dict()
      • TextLogger.on_train_start()
      • TextLogger.on_train_end()
      • TextLogger.on_sanity_check_start()
      • TextLogger.on_sanity_check_end()
      • TextLogger.on_exception()
      • TextLogger.on_train_epoch_start()
      • TextLogger.on_train_epoch_end()
      • TextLogger.on_validation_epoch_end()
      • TextLogger.on_validation_epoch_start()
    • default_save_package()
    • LightningTelemetry
      • LightningTelemetry.add_argparse_args()
      • LightningTelemetry.setup()
      • LightningTelemetry._after_initialization()
      • LightningTelemetry.on_fit_start()
      • LightningTelemetry.on_fit_end()
      • LightningTelemetry.on_exception()
      • LightningTelemetry._dump()

Submodules

• geowatch.utils.lightning_ext._jsonargparse_ext_ge_4_21_and_lt_4_22 module
• geowatch.utils.lightning_ext._jsonargparse_ext_ge_4_22_and_lt_4_24 module
• geowatch.utils.lightning_ext._jsonargparse_ext_ge_4_24_and_lt_4_xx module
• geowatch.utils.lightning_ext._jsonargparse_ext_ge_4_xx_and_lt_4_21 module
  • extract_scriptconfig_params()
  • extract_scriptconfig_param_args()
  • ArgumentParserPatches
    • ArgumentParserPatches.add_subclass_arguments()
    • ArgumentParserPatches._add_signature_arguments()
    • ArgumentParserPatches._add_signature_parameter()
    • ArgumentParserPatches._apply_actions()
  • _find_action_and_subcommand()
  • _action_aliases()
• 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()
  • _gpus_allowed_type()
  • _int_or_float_type()
  • 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
  • LightningArgumentParser_Extension
  • 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()
  • _sanitize_gpu_ids()
  • _normalize_parse_gpu_string_input()
  • _normalize_parse_gpu_input_to_list()
  • _get_all_available_gpus()
  • _check_unique()
  • _check_data_type()
  • _parse_gpu_ids()
• geowatch.utils.lightning_ext.util_device module
  • coerce_devices()
  • _test_lightning_is_sane()
  • _test_coerce_is_sane()
• 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