MCPcopy
hub / github.com/deepspeedai/DeepSpeed / __init__

Method __init__

deepspeed/runtime/eigenvalue.py:15–37  ·  view source on GitHub ↗
(self,
                 verbose=False,
                 max_iter=100,
                 tol=1e-2,
                 stability=0,
                 gas_boundary_resolution=1,
                 layer_name='',
                 layer_num=0)

Source from the content-addressed store, hash-verified

13class Eigenvalue(object):
14
15 def __init__(self,
16 verbose=False,
17 max_iter=100,
18 tol=1e-2,
19 stability=0,
20 gas_boundary_resolution=1,
21 layer_name='',
22 layer_num=0):
23 super().__init__()
24
25 self.verbose = verbose
26 self.max_iter = max_iter
27 self.tol = tol
28 self.stability = stability
29 self.gas_boundary_resolution = gas_boundary_resolution
30 self.layer_name = layer_name
31 self.layer_num = layer_num
32
33 assert len(self.layer_name) > 0 and layer_num > 0
34
35 log_dist(
36 f'enabled eigenvalue with verbose={verbose}, max_iter={max_iter}, tol={tol}, stability={stability}, gas_boundary_resolution={gas_boundary_resolution}, layer_name={layer_name}, layer_num={layer_num}',
37 ranks=[0])
38
39 # Replace all nan/pos-inf/neg-inf to zero
40 def nan_to_num(self, x):

Callers

nothing calls this directly

Calls 1

log_distFunction · 0.90

Tested by

no test coverage detected