diff --git a/README.rst b/README.rst index 13eb9343088fe41593bd6c381cecd44ce4884076..3ecb2322e642088db9ca688678e565d343c97e5c 100644 --- a/README.rst +++ b/README.rst @@ -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_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 + 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/