56f858ed1a
[tune][docs/util] gputil check, docs ( #11260 )
...
Co-authored-by: Amog Kamsetty <amogkam@users.noreply.github.com >
2020-10-10 00:54:31 -07:00
508cfa3540
[tune] Support yield and return statements ( #10857 )
...
* Support `yield` and `return` statements in Tune trainable functions
* Support anonymous metric with ``tune.report(value)``
* Raise on invalid return/yield value
* Fix end to end reporter test
2020-09-17 20:18:35 -07:00
d3feb83053
[tune] check for running session ( #10840 )
2020-09-16 18:55:11 -07:00
b5068d08bf
[tune] Fix restoration for function API PBT ( #9853 )
2020-08-03 12:36:17 -07:00
0c3b9ebeef
[tune/sgd] Document func_trainable and add checkpoint context ( #9739 )
...
Co-authored-by: krfricke <krfricke@users.noreply.github.com >
Co-authored-by: Amog Kamsetty <amogkam@users.noreply.github.com >
2020-07-30 09:46:37 -07:00
f3fdb5c5db
[tune] distributed torch wrapper ( #9550 )
...
* changes
* add-working
* checkpoint
* ccleanu
* fix
* ok
* formatting
* ok
* tests
* some-good-stuff
* fix-torch
* ddp-torch
* torch-test
* sessions
* add-small-test
* fix
* remove
* gpu-working
* update-tests
* ok
* try-test
* formgat
* ok
* ok
2020-07-26 09:37:22 -07:00
d35f0e40d0
[tune] Use public methods for trainable ( #9184 )
2020-07-01 11:00:00 -07:00
6c49c01837
[tune] Function API checkpointing ( #8471 )
...
Co-authored-by: krfricke <krfricke@users.noreply.github.com >
2020-06-15 10:42:54 -07:00
67c01455fe
[tune] tune.track -> tune.report ( #8388 )
2020-05-16 12:55:08 -07:00
Richard Liaw