diff --git a/aimodel/src/parse_args.py b/aimodel/src/parse_args.py index 1644a6b..c7c6053 100644 --- a/aimodel/src/parse_args.py +++ b/aimodel/src/parse_args.py @@ -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 --help. """) diff --git a/aimodel/src/subcommands/train_mono_predict.py b/aimodel/src/subcommands/train_mono_predict.py index 3b6d0bf..ed0ab74 100644 --- a/aimodel/src/subcommands/train_mono_predict.py +++ b/aimodel/src/subcommands/train_mono_predict.py @@ -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