deepmd.pt.optimizer.LKF#Classes#LKFOptimizerFunctions#distribute_indices(total_length, num_workers)Module Contents#deepmd.pt.optimizer.LKF.distribute_indices(total_length, num_workers)[source]#class deepmd.pt.optimizer.LKF.LKFOptimizer(params, kalman_lambda=0.98, kalman_nue=0.9987, block_size=5120)[source]#Bases: torch.optim.optimizer.Optimizer _params[source]# _state[source]# dist_init[source]# rank[source]# dindex = [][source]# remainder = 0[source]# __init_P() → None[source]# __get_blocksize()[source]# __get_nue()[source]# __split_weights(weight)[source]# __update(H, error, weights) → None[source]# set_grad_prefactor(grad_prefactor) → None[source]# step(error) → None[source]# get_device_id(index)[source]#
