From 3c1aef5913571259cf31a99a1bbb85b8fa68bd27 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Tue, 13 Sep 2022 19:36:56 +0100 Subject: [PATCH] update help --- aimodel/src/parse_args.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/aimodel/src/parse_args.py b/aimodel/src/parse_args.py index 957ca86..3ed7a13 100644 --- a/aimodel/src/parse_args.py +++ b/aimodel/src/parse_args.py @@ -17,7 +17,9 @@ This program trains, manipulates, visualises, and manages a contrastive learning It functions by first finding relationships between the rainfall radar data and the water depth + heightmap data (the 'pretrain' subcommand). After this, a decoder model to predict water depth (modelled as an image segmentation task), can then be trained. Available subcommands: - pretrain Pretrain a contrastive learning model as an encoder. + 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. For more information, do src/index.py --help. """)