sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-22 17:23:01 +00:00
Code Issues Projects Releases Packages Wiki Activity

update cli docs

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-11-24 16:38:07 +00:00
parent 773944f9fa
commit 011e0aef78
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
2 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
aimodel/src/parse_args.py
View file

@ -23,6 +23,7 @@ Available subcommands:
train Train an image segmentation head on the output of pretrain-predict. YOU MUST TRAIN A CONTRASTIVE LEARNING MODEL FIRST. 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. train-predict Make predictions using a model trained through the train subcommand.
train-mono Train a mono rainfall → water depth model. train-mono Train a mono rainfall → water depth model.
train-mono-predict Make predictions using a model trained through the train-mono subcommand.
For more information, do src/index.py <subcommand> --help. For more information, do src/index.py <subcommand> --help.
""") """)

2
aimodel/src/subcommands/train_mono_predict.py
View file

@ -71,7 +71,7 @@ def run(args):
sys.stderr.write(f"\n\n>>> This is TensorFlow {tf.__version__}\n\n\n") sys.stderr.write(f"\n\n>>> This is TensorFlow {tf.__version__}\n\n\n")
# Note that if using a directory of input files, the output order is NOT GUARANTEED TO BE THE SAME. In fact, it probably won't be. # Note that if using a directory of input files, the output order is NOT GUARANTEED TO BE THE SAME. In fact, it probably won't be (see dataset_mono for more details).
dataset = dataset_predict( dataset = dataset_predict(
dirpath_input=args.input, dirpath_input=args.input,
parallel_reads_multiplier=args.read_multiplier parallel_reads_multiplier=args.read_multiplier