search

ITMO-NSS-team / torch_DE_solver

Combines power of torch, numerical methods to conquer and solve ALL {O,P}DEs
https://torch-de-solver.readthedocs.io
BSD 3-Clause "New" or "Revised" License
59 stars 9 forks source link

Mixed precision #37

Closed nikiniki1 closed 1 year ago

nikiniki1 commented 1 year ago

Added PyTorch automatic mixed precision for both CPU/GPU + example. LBFGS is not currently supported due to PyTorch limitations. This mode allows you to quickly obtain a solution while sacrificing accuracy. An additional method can be tensor decomposition of the model parameters. Currently, experiments are being conducted with tensor decomposition (tensor-train). PR closes the issue #27.