PipelineDefinition: resources/definition/training_pipeline.py
TestName: fsx-mnist-training
Timeout: 3600
ExpectedTrainingImage: ((KMEANS_REGISTRY)).dkr.ecr.((REGION)).amazonaws.com/kmeans:1
Arguments:
  region: ((REGION))
  image: ((KMEANS_REGISTRY)).dkr.ecr.((REGION)).amazonaws.com/kmeans:1
  training_input_mode: File
  hyperparameters:
    k: "10"
    feature_dim: "784"
  channels:
    - ChannelName: train
      DataSource:
        FileSystemDataSource:
          FileSystemType: FSxLustre
          FileSystemAccessMode: ro
          FileSystemId: ((FSX_ID))
          DirectoryPath: /fsx/mnist_kmeans_example/input
      CompressionType: None
      ContentType: text/csv;label_size=0
      RecordWrapperType: None
      InputMode: File
  vpc_security_group_ids: ((FSX_SECURITY_GROUP))
  vpc_subnets: ((FSX_SUBNET))
  instance_type: ml.m5.xlarge
  instance_count: 1
  volume_size: 50
  max_run_time: 3600
  model_artifact_path: s3://((DATA_BUCKET))/mnist_kmeans_example/output
  network_isolation: "True"
  traffic_encryption: "False"
  spot_instance: "False"
  max_wait_time: 3600
  checkpoint_config: "{}"
  role: ((SAGEMAKER_ROLE_ARN))
