Struct TrainArgs 

pub struct TrainArgs {

    pub model: ModelKind,
    pub backend: BackendKind,
    pub max_boxes: usize,
    pub lambda_box: f32,
    pub lambda_obj: f32,
    pub input_source: TrainingInputSource,
    pub warehouse_manifest: String,
    pub dataset_root: String,
    pub labels_subdir: String,
    pub images_subdir: String,
    pub epochs: usize,
    pub batch_size: usize,
    pub lr: f32,
    pub infer_obj_thresh: f32,
    pub infer_iou_thresh: f32,
    pub checkpoint_out: Option<String>,
}

Fields

model: ModelKind

Model to train.

backend: BackendKind

Backend to use (ndarray or wgpu if enabled).

max_boxes: usize

Maximum boxes per image (pads/truncates to this for training).

lambda_box: f32

Loss weight for box regression.

lambda_obj: f32

Loss weight for objectness.

input_source: TrainingInputSource

Training input source (warehouse by default).

warehouse_manifest: String

Warehouse manifest path (used with –input-source warehouse).

dataset_root: String

Capture-log dataset root containing labels/ and images/.

labels_subdir: String

Labels subdirectory relative to dataset root (capture-logs only).

images_subdir: String

Images subdirectory relative to dataset root (capture-logs only).

epochs: usize

Number of epochs.

batch_size: usize

Batch size.

lr: f32

Learning rate.

infer_obj_thresh: f32

Objectness threshold (for future eval).

infer_iou_thresh: f32

IoU threshold (for future eval).

checkpoint_out: Option<String>

Checkpoint output path (defaults by model if not provided).

Trait Implementations

impl Args for TrainArgs

fn group_id() -> Option<Id>

Report the ArgGroup::id for this set of arguments

fn augment_args<'b>(__clap_app: Command) -> Command

Append to Command so it can instantiate Self via FromArgMatches::from_arg_matches_mut

fn augment_args_for_update<'b>(__clap_app: Command) -> Command

Append to Command so it can instantiate self via FromArgMatches::update_from_arg_matches_mut

impl CommandFactory for TrainArgs

fn command<'b>() -> Command

Build a Command that can instantiate Self.

fn command_for_update<'b>() -> Command

Build a Command that can update self.

impl Debug for TrainArgs

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl FromArgMatches for TrainArgs

fn from_arg_matches(__clap_arg_matches: &ArgMatches) -> Result<Self, Error>

Instantiate Self from ArgMatches, parsing the arguments as needed.

fn from_arg_matches_mut( __clap_arg_matches: &mut ArgMatches, ) -> Result<Self, Error>

Instantiate Self from ArgMatches, parsing the arguments as needed.

fn update_from_arg_matches( &mut self, __clap_arg_matches: &ArgMatches, ) -> Result<(), Error>

Assign values from ArgMatches to self.

fn update_from_arg_matches_mut( &mut self, __clap_arg_matches: &mut ArgMatches, ) -> Result<(), Error>

Assign values from ArgMatches to self.

impl Parser for TrainArgs

fn parse() -> Self

Parse from std::env::args_os(), exit on error.

fn try_parse() -> Result<Self, Error>

Parse from std::env::args_os(), return Err on error.

fn parse_from<I, T>(itr: I) -> Self

where I: IntoIterator<Item = T>, T: Into<OsString> + Clone,

Parse from iterator, exit on error.

fn try_parse_from<I, T>(itr: I) -> Result<Self, Error>

where I: IntoIterator<Item = T>, T: Into<OsString> + Clone,

Parse from iterator, return Err on error.

fn update_from<I, T>(&mut self, itr: I)

where I: IntoIterator<Item = T>, T: Into<OsString> + Clone,

Update from iterator, exit on error.

fn try_update_from<I, T>(&mut self, itr: I) -> Result<(), Error>

where I: IntoIterator<Item = T>, T: Into<OsString> + Clone,

Update from iterator, return Err on error.