update cli docs

This commit is contained in:
Starbeamrainbowlabs 2022-11-24 16:38:07 +00:00
parent 773944f9fa
commit 011e0aef78
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
2 changed files with 2 additions and 1 deletions

View file

@ -23,6 +23,7 @@ Available subcommands:
train Train an image segmentation head on the output of pretrain-predict. YOU MUST TRAIN A CONTRASTIVE LEARNING MODEL FIRST.
train-predict Make predictions using a model trained through the train subcommand.
train-mono Train a mono rainfall water depth model.
train-mono-predict Make predictions using a model trained through the train-mono subcommand.
For more information, do src/index.py <subcommand> --help.
""")

View file

@ -71,7 +71,7 @@ def run(args):
sys.stderr.write(f"\n\n>>> This is TensorFlow {tf.__version__}\n\n\n")
# Note that if using a directory of input files, the output order is NOT GUARANTEED TO BE THE SAME. In fact, it probably won't be.
# Note that if using a directory of input files, the output order is NOT GUARANTEED TO BE THE SAME. In fact, it probably won't be (see dataset_mono for more details).
dataset = dataset_predict(
dirpath_input=args.input,
parallel_reads_multiplier=args.read_multiplier