torch_em.util.training

@private

View Source

 1"""@private
 2"""
 3import argparse
 4
 5
 6def parser_helper(description=None, default_iterations=int(1e5), default_batch_size=1, require_input=True):
 7    description = "Run torch_em training" if description is None else description
 8    parser = argparse.ArgumentParser(description)
 9    if require_input:
10        parser.add_argument("-i", "--input", required=True,
11                            help="Path to the input data, if not present an attempt will be made to download the data.")
12    parser.add_argument("-n", "--n_iterations", type=int, default=default_iterations,
13                        help="The number of training iterations.")
14    parser.add_argument("-b", "--batch_size", type=int, default=default_batch_size,
15                        help="The batch size")
16    parser.add_argument("--check", "-c", type=int, default=0, help="Check the data loader instead of running training.")
17    parser.add_argument("--from_checkpoint", type=int, default=0, help="Start training from existing checkpoint.")
18    parser.add_argument("--device", type=str, default=None)
19    return parser

def parser_helper( description=None, default_iterations=100000, default_batch_size=1, require_input=True): View Source

 7def parser_helper(description=None, default_iterations=int(1e5), default_batch_size=1, require_input=True):
 8    description = "Run torch_em training" if description is None else description
 9    parser = argparse.ArgumentParser(description)
10    if require_input:
11        parser.add_argument("-i", "--input", required=True,
12                            help="Path to the input data, if not present an attempt will be made to download the data.")
13    parser.add_argument("-n", "--n_iterations", type=int, default=default_iterations,
14                        help="The number of training iterations.")
15    parser.add_argument("-b", "--batch_size", type=int, default=default_batch_size,
16                        help="The batch size")
17    parser.add_argument("--check", "-c", type=int, default=0, help="Check the data loader instead of running training.")
18    parser.add_argument("--from_checkpoint", type=int, default=0, help="Start training from existing checkpoint.")
19    parser.add_argument("--device", type=str, default=None)
20    return parser