File size: 225 Bytes
6aa5e9b
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
# @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