diff --git a/aimodel/src/parse_args.py b/aimodel/src/parse_args.py index 87eacaa..fec8e8a 100644 --- a/aimodel/src/parse_args.py +++ b/aimodel/src/parse_args.py @@ -21,6 +21,7 @@ Available subcommands: pretrain-predict Make predictions using a trained contrastive learning encoder. pretrain-plot Plot using embeddings predicted using pretrain-predict. 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. For more information, do src/index.py --help. """)