topk

Module Contents

TopkCompressor

Compressor for federated communication

class TopkCompressor(compress_ratio)

Bases: fedlab.contrib.compressor.compressor.Compressor

Compressor for federated communication Top-k gradient or weights selection :param compress_ratio: compress ratio :type compress_ratio: float

compress(tensor)

compress tensor into (values, indices) :param tensor: tensor :type tensor: torch.Tensor

Returns:

(values, indices)

Return type:

tuple

decompress(values, indices, shape)

decompress tensor