Skip to content

Releases: lixilinx/psgd_torch

archived code

17 Dec 06:39
e1a82fc
Compare
Choose a tag to compare
Update preconditioned_stochastic_gradient_descent.py

1, replace trtrs with triangular_solve due to torch's API update

2, use torch.chain_matmul for things like A @ B @ C ...