mirror of
https://github.com/sbrl/research-rainfallradar
synced 2024-11-04 17:13:02 +00:00
update cli docs
This commit is contained in:
parent
773944f9fa
commit
011e0aef78
2 changed files with 2 additions and 1 deletions
|
@ -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-predict Make predictions using a model trained through the train subcommand.
|
||||
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.
|
||||
""")
|
||||
|
|
|
@ -71,7 +71,7 @@ def run(args):
|
|||
|
||||
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(
|
||||
dirpath_input=args.input,
|
||||
parallel_reads_multiplier=args.read_multiplier
|
||||
|
|
Loading…
Reference in a new issue