Skip to content

archived code

Latest
Compare
Choose a tag to compare
@lixilinx lixilinx released this 17 Dec 06:39
· 116 commits to master since this release
e1a82fc
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 ...