()
| 25 | |
| 26 | |
| 27 | def para_args(): |
| 28 | parser = argparse.ArgumentParser(description="Network configurations!") |
| 29 | parser.add_argument("--config-network", default="config/network/spherevlad.yaml", metavar="FILE", |
| 30 | help="path to config file", type=str) |
| 31 | parser.add_argument("--config-dataset", default="config/dataset/PITT.yaml", metavar="FILE", |
| 32 | help="path to config file", type=str) |
| 33 | args = parser.parse_args() |
| 34 | return args |
| 35 | |
| 36 | |
| 37 | def train(config, logger, neptune): |