I don’t want this log display in stdout:
INFO train_loop_utils.py:617 -- Data type <class 'int'> doesn't support being moved to device.
So I add this code:
level = logging.WARNING
logger = logging.getLogger("ray")
logger.setLevel(level)
stream_handler = logging.StreamHandler(sys.stdout)
stream_handler.setLevel(level)
logger.addHandler(stream_handler)
I test add this code before or after ray.init()
, neither of them works.
I also tested ray.init(logging_level="warning")
, which didn’t work either.