examples/multimodal/mini_dalle/pytorch/benchmarks.yml at master · graphcore/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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
---
common_options: &common_options
  data:
    throughput:
      reduction_type: "final"
      regexp: "throughput: *(.*?) samples\/sec"
    loss:
      reduction_type: "final"
      regexp: 'loss: *(\d*\.\d*)'
    compile_time:
      regexp: "Compiled model in *(.*?) secs"
  output:
    - [samples/sec, 'throughput']
    - [loss, 'loss']
    - [sec, 'compile_time']

config_options: &config_options
  requirements_path: requirements.txt

pytorch_minidalle_train_real_pod4:
  <<: [*common_options, *config_options]
  cmd: >-
    python3 train.py
      --config L16
      --checkpoint-output-dir ""
      --byteio True
      --input-folder $DATASETS_DIR/coco
      --epochs 2

pytorch_minidalle_train_real_pod16:
  <<: [*common_options, *config_options]
  cmd: >-
    python3 train.py
      --config L16_POD16
      --checkpoint-output-dir ""
      --byteio True
      --input-folder $DATASETS_DIR/coco
      --epochs 2

pytorch_minidalle_train_real_pod64:
  <<: [*common_options, *config_options]
  cmd: >-
    python3 train.py
      --config L16_POD64
      --checkpoint-output-dir ""
      --byteio True
      --input-folder $DATASETS_DIR/coco
      --epochs 2

pytorch_minidalle_train_real_pod64_conv:
  <<: [*common_options, *config_options]
  cmd: >-
    python3 train.py
      --config L16_POD64
      --byteio True
      --input-folder $DATASETS_DIR/coco
      --wandb
      --wandb-project-name torch-miniDALL-E
      --wandb-run-name pytorch_minidalle_train_real_pod64_conv
      --checkpoint-output-dir output/ckpt