mirror of
https://github.com/aimingmed/aimingmed-ai.git
synced 2026-01-30 02:59:42 +08:00
44 lines
1006 B
Python
44 lines
1006 B
Python
#!/usr/bin/env python
|
|
"""
|
|
{{cookiecutter.long_description}}
|
|
"""
|
|
import argparse
|
|
import logging
|
|
import wandb
|
|
|
|
|
|
logging.basicConfig(level=logging.INFO, format="%(asctime)-15s %(message)s")
|
|
logger = logging.getLogger()
|
|
|
|
|
|
def go(args):
|
|
|
|
run = wandb.init(job_type="{{cookiecutter.job_type}}")
|
|
run.config.update(args)
|
|
|
|
# Download input artifact. This will also log that this script is using this
|
|
# particular version of the artifact
|
|
# artifact_local_path = run.use_artifact(args.input_artifact).file()
|
|
|
|
######################
|
|
# YOUR CODE HERE #
|
|
######################
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
parser = argparse.ArgumentParser(description="{{cookiecutter.short_description}}")
|
|
|
|
{% for arg_name in cookiecutter.parameters.split(",") %}
|
|
parser.add_argument(
|
|
"--{{arg_name}}",
|
|
type=## INSERT TYPE HERE: str, float or int,
|
|
help=## INSERT DESCRIPTION HERE,
|
|
required=True
|
|
)
|
|
{% endfor %}
|
|
|
|
args = parser.parse_args()
|
|
|
|
go(args)
|