[tune] Tune onto Logging Module (#2882)

Moves Tune onto logging in Python. Ignores examples and tests.
This commit is contained in:
Richard Liaw
2018-09-16 12:09:36 -07:00
committed by GitHub
parent a8248e8628
commit f372f48bf3
17 changed files with 121 additions and 63 deletions
+5 -1
View File
@@ -6,6 +6,7 @@ from datetime import datetime
import gzip
import io
import logging
import os
import pickle
import shutil
@@ -19,6 +20,8 @@ from ray.tune.result import (DEFAULT_RESULTS_DIR, TIME_THIS_ITER_S,
TIMESTEPS_THIS_ITER, DONE, TIMESTEPS_TOTAL)
from ray.tune.trial import Resources
logger = logging.getLogger(__name__)
class Trainable(object):
"""Abstract class for trainable models, functions, etc.
@@ -231,7 +234,8 @@ class Trainable(object):
"data": data,
})
if len(compressed) > 10e6: # getting pretty large
print("Checkpoint size is {} bytes".format(len(compressed)))
logger.info("Checkpoint size is {} bytes".format(
len(compressed)))
f.write(compressed)
shutil.rmtree(tmpdir)