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

Update README.rst

parent 92584787
No related branches found
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