deepfold.train.gradient_clipping.AsyncGradientClipping¶

class deepfold.train.gradient_clipping.AsyncGradientClipping(device: device, comm_group: ProcessGroup | None = None, norm_type: float = 2.0)[source]¶

__init__(device: device, comm_group: ProcessGroup | None = None, norm_type: float = 2.0) → None[source]¶

Methods

`__init__`(device[, comm_group, norm_type])
`get_clip_scale`(max_norm[, eps])