11 lines
424 B
JSON
11 lines
424 B
JSON
{
|
|
"train_component.yaml": {
|
|
"implementation.container.command": ["python3", "cifar10/cifar10_pytorch.py"],
|
|
"implementation.container.image": "public.ecr.aws/pytorch-samples/kfp_samples:latest"
|
|
},
|
|
"preprocess_component.yaml": {
|
|
"implementation.container.command": ["python3", "cifar10/cifar10_pre_process.py"],
|
|
"implementation.container.image": "public.ecr.aws/pytorch-samples/kfp_samples:latest"
|
|
}
|
|
|
|
} |