mlcube_examples/emdenoise/mlcube.yaml at master · mlcommons/mlcube_examples · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
name: emdenoise
description: MLCommons EMDenoise MLCube example
authors:
  - {name: "First Second", email: "first.second@company.com", org: "Company Inc."}

platform:
  accelerator_count: 0

docker:
  image: mlcommons/sciml_em_denoise:0.0.1

tasks:
  download:
    parameters:
      outputs:
        raw_data_dir: {type: directory, default: raw_data}
        log_dir: {type: directory, default: logs}
  preprocess:
    parameters:
      inputs:
        raw_data_dir: {type: directory, default: raw_data}
      outputs:
        data_dir: {type: directory, default: data}
        log_dir: {type: directory, default: logs}
  train:
    parameters:
      inputs:
        data_dir: {type: directory, default: data}
        parameters_file: {type: file, default: parameters/default.parameters.yaml}
      outputs:
        model_dir: {type: directory, default: model}
        output_dir: {type: directory, default: output}
        log_dir: {type: directory, default: logs}
  test:
    parameters:
      inputs:
        data_dir: {type: directory, default: data}
        model_dir: {type: directory, default: model}
        parameters_file: {type: file, default: parameters/default.parameters.yaml}
      outputs:
        output_dir: {type: directory, default: output}
        log_dir: {type: directory, default: logs}