diff --git a/aimodel/src/parse_args.py b/aimodel/src/parse_args.py index 162be03..87eacaa 100644 --- a/aimodel/src/parse_args.py +++ b/aimodel/src/parse_args.py @@ -20,6 +20,7 @@ Available subcommands: pretrain Pretrain a contrastive learning model as an encoder. 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. For more information, do src/index.py --help. """)