Skip to content
Snippets Groups Projects
Select Git revision
  • 29d59f972da9828efcf7a1e2bd3c981342a30bcc
  • master default
  • develop
  • map_stuff
  • bullet
  • player_stuff
  • enemy
  • login_page
  • mob_stuff
  • register_page
10 results

functional-red-black-tree

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

Recommend running everything from the project root directory (ensures yolo will work).

Main sparse coding file is found at:
sparse_coding_torch/train_sparse_model.py
Sparse coding command example:
python sparse_coding_torch/train_sparse_model.py --dataset pnb --train
Main classification model is found at:
sparse_coding_torch/train_classifier.py
Classification command example:
python sparse_coding_torch/train_classifier.py --sparse_checkpoint sparse.pt --dataset pnb --train
PTX Checkpoints can be found at:
ptx_tensorflow/