wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-04 15:57:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] Add PyTorch MNIST Example + Misc. Tweaks (#2708)

Browse Source
This commit is contained in:
Richard Liaw
2018-08-30 16:18:56 -07:00
committed by GitHub
parent 224d38cbb2
commit 0347e6418b
8 changed files with 492 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/function_runner.py
+6 -5
View File
@@ -115,11 +115,12 @@ class FunctionRunner(Trainable):
time.sleep(1)
result = self._status_reporter._get_and_clear_status()
curr_ts_total = result.get(TIMESTEPS_TOTAL,
self._last_reported_timestep)
result.update(
timesteps_this_iter=(curr_ts_total - self._last_reported_timestep))
self._last_reported_timestep = curr_ts_total
curr_ts_total = result.get(TIMESTEPS_TOTAL)
if curr_ts_total is not None:
result.update(
timesteps_this_iter=(
curr_ts_total - self._last_reported_timestep))
self._last_reported_timestep = curr_ts_total
return result