PR #711 Add early stopping to flax.training

Add early stopping to flax.training #711

copybara-service merged 13 commits into google:master from master

google-cla added cla: yes

jheek commented on 2020-12-08

jheek added Status: blocked

jheek added Type: enhancement

avital added this to the Non-core Flax Improvements milestone 5 years ago

gmittal requested a review from

jheek 5 years ago

Add early stopping module

bc03612a

add EOF

bc2f1615

fix style

2666b87d

fix

dfa79bf6

update style

27d68ced

Update early_stopping.py

22aa1d75

Add basic tests.

0931fac0

Add reuse

84dd7dd1

add EOF

cae2aa8b

Change ES to dataclass

3071c033

Respect types

bac44041

Update replace

bbbab8c8

avital removed Type: enhancement

change to pytreenode

f938c9fc

BertrandRdp added Priority: P2 - no schedule

jheek commented on 2021-02-03

jheek requested a review from

jheek 4 years ago

jheek approved these changes on 2021-04-08

jheek added pull ready

jheek removed Status: blocked

copybara-service merged 6faf6c1a into master 4 years ago

Reviewers

jheek

Assignees

No one assigned

Labels

cla: yes Priority: P2 - no schedule pull ready

Milestone

No milestone

flax Add early stopping to flax.training #711 Merged

Add early stopping to flax.training #711

flax
Add early stopping to flax.training
#711

Merged