Jingwei Huang, Shan Huang, and Mingwei Sun. DeepLM: Large-scale Nonlinear Least Squares on Deep Learning Frameworks using Stochastic Domain Decomposition, CVPR 2021
Please install the following
- pytorch.
- OpenMP (optional)
Then, run the example via
sh example.sh
A full set of test data can be downloaded from the BAL page (Software & Data -> Available Dataset).
Please check the examples folder for more application and features.
© 2021 Jingwei Huang All Rights Reserved
IMPORTANT: If you use this code please cite the following in any resulting publication:
@inproceedings{huang2021deeplm,
title={DeepLM: Large-scale Nonlinear Least Squares on Deep Learning Frameworks using Stochastic Domain Decomposition},
author={Huang, Jingwei and Huang, Shan and Sun, Mingwei},
booktitle={Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition},
pages={10308--10317},
year={2021}
}