Location via proxy:   [ UP ]  
[Report a bug]   [Manage cookies]                
Skip to content

PyTorch and Torch implementation for our accepted CVPR 2020 paper (Oral): Controllable Orthogonalization in Training DNNs

License

Notifications You must be signed in to change notification settings

huangleiBuaa/ONI

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

19 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Orthogonalization by Newton's Iteration (ONI)

Code for reproducing the results of our accpeted CVPR 2020 paper (Oral): Controllable Orthogonalization in Training DNNs (arXiv:2004.00917)

This repo provides the Pytorch and Torch implementations of our methods.

  • ONI_PyTorch: Pytorch project for ImageNet classification and GAN experiments.
  • ONI_Torch: Torch project for Cifar-10 Classfication.

Citations

If you find this repo benefits your research, please consider citing:

@inproceedings{2020_CVPR_Huang,
  author    = {Lei Huang and Li Liu and Fan Zhu and Diwen Wan and Zehuan Yuan and Bo Li and Ling Shao},
  title     = {Controllable Orthogonalization in Training DNNs},
  booktitle = {IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
  year      = {2020},
  }

Contact

Email: huangleig@nlsde.buaa.edu.cn.

About

PyTorch and Torch implementation for our accepted CVPR 2020 paper (Oral): Controllable Orthogonalization in Training DNNs

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published