Ross Wightman
|
42c1f0cf6c
|
Fix lars tests
|
2021-08-18 21:05:34 -07:00 |
|
Ross Wightman
|
a426511c95
|
More optimizer cleanup. Change all to no longer use .data. Improve (b)float16 use with adabelief. Add XLA compatible Lars.
|
2021-08-18 17:21:56 -07:00 |
|
Ross Wightman
|
a6af48be64
|
add madgradw optimizer
|
2021-08-17 22:19:27 -07:00 |
|
Ross Wightman
|
55fb5eedf6
|
Remove experiment from lamb impl
|
2021-08-17 21:48:26 -07:00 |
|
Ross Wightman
|
959eaff121
|
Add optimizer tests and update testing to pytorch 1.9
|
2021-08-17 17:59:15 -07:00 |