Skip to content

dnn cli #25

@brettviren

Description

@brettviren

We want to capture workflow for training and validating DNNROI and future deep learning components in the WC ecosystem so that they may be (re)exercised in the future and extended to new detectors.

To do that we want a CLI which can perform tasks in the workflow for existing DL (DNNROI) and future models in a general, extensible way.

This issue holds an outline which may evolve over the course of development. We want these features:

  • a train subcommand to exercise training on the named network
  • a plot subcommand to plot training metadata and app-specific plots
  • implementation of U-Net and the variant that DNNROI uses
  • a general abstract API (Python module pattern) for each application with general base imp and derived imp for DNNROI
    • network constructor
    • dataset and any transformations
    • training loop
    • training metric plots
    • model specific validation plots
  • tracking and plotting of training metadata
  • persist and load the model

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions