deeprm train

Tools for preparing data and training models. Torch is optional at install time; imported only when needed.

Group help

DeepRM Training Module

usage: deeprm train [-h] {prep,run,compile} ...

Positional Arguments

{prep,run,compile}

Possible choices: prep, compile, run

Sub-commands

prep

Prepare training data

deeprm train prep

compile

Compile training dataset

deeprm train compile

run

Launch training

deeprm train run

Examples

# Prepare training data
deeprm train prep --in train_raw/ --out train_prep/

# Compile dataset shards
deeprm train compile --in train_prep/ --out ds/

# Launch training
deeprm train run --config configs/train.yaml --out runs/exp1