Module epiclass.epiatlas_training_no_valid

Main for training on EpiAtlas data without a validation set.

Functions

def main()

main called from command line, edit to change behavior

def parse_arguments() ‑> argparse.Namespace

argument parser for command line

def train_without_valid(my_data: DataSet, hparams: Dict, logger: pl_loggers.CometLogger, restore: bool) ‑> None

Wrapper for convenience. Skip training if restore is True

Classes

class DatasetError (*args: object)

Custom error

Expand source code

class DatasetError(Exception):
    """Custom error"""

    def __init__(self, *args: object) -> None:
        print(
            "\n--- ERROR : Verify source files, filters, and min_class_size. ---\n",
            file=sys.stderr,
        )
        super().__init__(*args)

Ancestors

• builtins.Exception
• builtins.BaseException