Skip to content

An efficient implementation of Successive Subspace Learning using PyTorch library.

Notifications You must be signed in to change notification settings

zohrehazizi/torch_SSL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

12 Commits
 
 
 
 
 
 

Repository files navigation

Succesive Subspace Learning (SSL)

An efficient implemenation of SSL in PyTorch library

This repo containes a very fast and effiecient implemenation of SSL using torch library.

Please cite the following reference if you use torch_SSL:

@article{azizi2022pager,
  title={PAGER: Progressive Attribute-Guided Extendable Robust Image Generation},
  author={Azizi, Zohreh and Kuo, C-C Jay and others},
  journal={APSIPA Transactions on Signal and Information Processing},
  volume={11},
  number={1},
  year={2022},
  publisher={Now Publishers, Inc.}
}

Usage

  • Clone the repo and cd into it.
  • Run pip install torch.
  • Run pip install numpy.
  • If you wish to force the code to run on CPU, open torch_configs.py and uncomment line #6. Otherwise, it will be run on GPU if available.
  • Run torch_ssl.py to see the example usage included under if __name__=="__main__":.

About

An efficient implementation of Successive Subspace Learning using PyTorch library.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages