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

# overfits to 3 batches

defaults:
  - default

trainer:
  max_epochs: 20
  overfit_batches: 3

# model ckpt and early stopping need to be disabled during overfitting
callbacks: null