profiler.yaml 225 Bytes
Newer Older
chenzk's avatar
v1.0  
chenzk committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
# @package _global_

# runs with execution time profiling

defaults:
  - default

trainer:
  max_epochs: 1
  # profiler: "simple"
  profiler: "advanced"
  # profiler: "pytorch"
  accelerator: gpu

  limit_train_batches: 0.02