ray.tune.utils.诊断序列化

ray.tune.utils.diagnose_serialization(trainable: Callable)

用于检测为什么您的可训练函数无法序列化的工具。

参数:

trainable – 传递给 tune.Tuner(trainable) 的可训练对象。目前仅支持函数 API。

返回:

布尔值 | 不可序列化的对象集合。

示例：

import threading
# this is not serializable
e = threading.Event()

def test():
    print(e)

diagnose_serialization(test)
# should help identify that 'e' should be moved into
# the `test` scope.

# correct implementation
def test():
    e = threading.Event()
    print(e)

assert diagnose_serialization(test) is True

开发者API: 此API可能会在Ray的次要版本之间发生变化。