diff --git a/README.rst b/README.rst index 13eb9343088fe41593bd6c381cecd44ce4884076..91108d43c9591d7971959f06df2f50dd3240103c 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 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/