Skip to content
Snippets Groups Projects
Select Git revision
  • 7933d6fdc7523be9fbafee4ef559ae5a85ff6b35
  • main default
  • final_pytorch
  • phase-1-final
4 results

sparse_coding_torch

user avatar
hannandarryl authored
7933d6fd
History

Sparse Coding Library for Tensorflow

Installation

To install the package, clone it to a local folder, then run pip install -e . from the folder to install the package in edit mode (so changes to the code are used).

Overview

Main sparse coding file is found at: sparse_coding_torch/train_sparse_model.py

Sparse coding command example: python train_sparse_model.py

Main classification model is found at: sparse_coding_torch/train_classifier.py

Classification command example: python small_data_classifier.py --checkpoint sparse.pt