spu设备的声明

[secretflow]

spu设备的声明

[wuxibin89]

sf.utils.testing.cluster_def是一个用来快速构建spu config的helper，你可以手动构建runtime_config，参见SPU RuntimeConfig。例如：

{
    'nodes': [
        {
            'party': 'alice',
            'id': 'local:0',
            'address': '127.0.0.1:9001',
            'listen_address': '' # Optional. Address will be used if listen_address is empty.
        },
        {
            'party': 'bob',
            'id': 'local:1',
            'address': '127.0.0.1:9002',
            'listen_address': ''
        },
    ],
    'runtime_config': {
        'protocol': spu.spu_pb2.SEMI2K,
        'field': spu.spu_pb2.FM128,
        'sigmoid_mode': spu.spu_pb2.RuntimeConfig.SIGMOID_REAL,
    }
}

[Wyyyb]

好的，感谢清晰明了的解答。