pipelines/samples/contrib/pytorch-samples/utils/template-generation.md

PyTorch component.yaml templates

The templates for generating component.yaml are placed under components/PyTorch/templates.

Link to component.yaml templates

component.yaml generation

There are two different ways to generate component.yaml files for all the components.

1. Generate templates using utility

The following utility has been created to generate templates dynamically during runtime. If there are no mapping scecified, the utility copies the templates into output directory (yaml folder) If the mappings are specified, the utility replaces the key, value pairs and generates component.yaml into yaml folder.

Run the following command to generate the templates

python utils/generate_templates.py cifar10/template_mapping.json

Sample mapping file is shown as below

{
  "minio_component.yaml": {
    "implementation.container.image": "public.ecr.aws/pytorch-samples/kfp_samples:latest
  }

}

The above mentioned mapping will replace the image name in minio_component.yaml

2. Manually editing the templates

When there are more changes in the templates and new key value pairs has to be introduced, it would be easier to manually edit the templates placed under components/PyTorch/templates

Once the templates are manually edited, simply run the template generation script.

python utils/generate_templates.py cifar10/template_mapping.json

The script will copy over the edited templates to output folder name yaml under pytorch-samples