From 3dee3d8908e76f79dec3293e3117e95603b33830 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Wed, 28 Sep 2022 16:23:47 +0100 Subject: [PATCH] update cli help --- 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 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. """)