MCPcopy Create free account
hub / github.com/deepspeedai/DeepSpeed / _configure_eigenvalue

Method _configure_eigenvalue

deepspeed/runtime/engine.py:2516–2527  ·  view source on GitHub ↗
(self)

Source from the content-addressed store, hash-verified

2514 return optimizer
2515
2516 def _configure_eigenvalue(self):
2517 eigenvalue = Eigenvalue(
2518 verbose=self.eigenvalue_verbose(),
2519 max_iter=self.eigenvalue_max_iter(),
2520 tol=self.eigenvalue_tol(),
2521 stability=self.eigenvalue_stability(),
2522 gas_boundary_resolution=self.eigenvalue_gas_boundary_resolution(),
2523 layer_name=self.eigenvalue_layer_name(),
2524 layer_num=self.eigenvalue_layer_num(),
2525 )
2526
2527 return eigenvalue
2528
2529 def _configure_progressive_layer_drop(self):
2530 pld = ProgressiveLayerDrop(theta=self.pld_theta(), gamma=self.pld_gamma())

Callers 1

__init__Method · 0.95

Calls 8

eigenvalue_verboseMethod · 0.95
eigenvalue_max_iterMethod · 0.95
eigenvalue_tolMethod · 0.95
eigenvalue_stabilityMethod · 0.95
eigenvalue_layer_nameMethod · 0.95
eigenvalue_layer_numMethod · 0.95
EigenvalueClass · 0.90

Tested by

no test coverage detected