Skip to content
Snippets Groups Projects
Commit 51972f78 authored by dwh48's avatar dwh48
Browse files

Update README.rst

parent 92584787
Branches
No related tags found
No related merge requests found
......@@ -10,15 +10,20 @@ To install the package, clone it to a local folder, then run `pip install -e .`
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 train_sparse_model.py
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 small_data_classifier.py --checkpoint sparse.pt
python sparse_coding_torch/train_classifier.py --sparse_checkpoint sparse.pt --dataset pnb --train
PTX Checkpoints can be found at:
ptx_tensorflow/
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment