flax
sst2 example: Switch to Optax optimizer and use TrainState.
#1306

Merged

sst2 example: Switch to Optax optimizer and use TrainState. #1306

copybara-service merged 4 commits into google:master from bastings:sst2_optax

google-cla added cla: yes

bastings requested a review from

marcvanzee 4 years ago

bastings requested a review from

andsteing 4 years ago

bastings force pushed 4 years ago

Switch to Optax optimizer and use TrainState.

a3990ed1

bastings force pushed to a3990ed1 4 years ago

andsteing dismissed these changes on 2021-05-06

andsteing assigned

andsteing 4 years ago

Merge remote-tracking branch 'upstream/master' into sst2_optax

cf27baf3

Move to use Optax optimizer.

afd14928

bastings dismissed their stale review 4 years ago

Thanks! Took care of comments, squashed commits.

bastings requested a review from

andsteing 4 years ago

bastings commented on 2021-05-07

andsteing approved these changes on 2021-05-07

pylint fixes

a64d88d5

andsteing added pull ready

copybara-service merged 48b5707a into master 4 years ago

bastings deleted the sst2_optax branch 4 years ago

Reviewers

andsteing

marcvanzee

Assignees

andsteing

Labels

cla: yes pull ready

Milestone

No milestone