cli package

Submodules

cli.common module

Common CLI functionality between models.

class cli.common.CommonCLI(*args, **kwargs)

Bases: LightningCLI

Common CLI functionality between models.

default_arguments: dict[str, Any] = {'model_checkpoint_dice_macro_class_2_3.auto_insert_metric_name': False, 'model_checkpoint_dice_macro_class_2_3.mode': 'max', 'model_checkpoint_dice_macro_class_2_3.monitor': 'val/dice_macro_class_2_3', 'model_checkpoint_dice_macro_class_2_3.save_last': False, 'model_checkpoint_dice_macro_class_2_3.save_top_k': 1, 'model_checkpoint_dice_macro_class_2_3.save_weights_only': True, 'model_checkpoint_dice_weighted.auto_insert_metric_name': False, 'model_checkpoint_dice_weighted.mode': 'max', 'model_checkpoint_dice_weighted.monitor': 'val/dice_weighted_avg', 'model_checkpoint_dice_weighted.save_last': False, 'model_checkpoint_dice_weighted.save_top_k': 1, 'model_checkpoint_dice_weighted.save_weights_only': True, 'model_checkpoint_last.auto_insert_metric_name': False, 'model_checkpoint_last.enable_version_counter': False, 'model_checkpoint_last.save_last': True, 'model_checkpoint_last.save_weights_only': True, 'model_checkpoint_val_loss.auto_insert_metric_name': False, 'model_checkpoint_val_loss.monitor': 'loss/val', 'model_checkpoint_val_loss.save_last': False, 'model_checkpoint_val_loss.save_top_k': 1, 'model_checkpoint_val_loss.save_weights_only': True, 'trainer.devices': 1}

multi_frame: bool = True

before_instantiate_classes() → None

Set the last checkpoint name.

add_arguments_to_parser(parser: LightningArgumentParser)

Set the default arguments and add the arguments to the parser.

class cli.common.I2RInternshipCommonCLI(*args, **kwargs)

Bases: CommonCLI

Internship project common CLI functionality.

add_arguments_to_parser(parser: LightningArgumentParser)

Set the default arguments and add the arguments to the parser.

Module contents

Common CLI functionality.