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 .` ...@@ -10,15 +10,20 @@ To install the package, clone it to a local folder, then run `pip install -e .`
Overview Overview
------------ ------------
Recommend running everything from the project root directory (ensures yolo will work).
Main sparse coding file is found at: Main sparse coding file is found at:
sparse_coding_torch/train_sparse_model.py sparse_coding_torch/train_sparse_model.py
Sparse coding command example: 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: Main classification model is found at:
sparse_coding_torch/train_classifier.py sparse_coding_torch/train_classifier.py
Classification command example: 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