From 4d8ae21a45589b56c25df69432a723867e6b810f Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 19 Oct 2022 17:31:42 +0100 Subject: [PATCH] update cli help text --- aimodel/src/parse_args.py | 1 + 1 file changed, 1 insertion(+) 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. """)