Skip to content
Snippets Groups Projects
Commit 4030cc2a authored by hannandarryl's avatar hannandarryl
Browse files

Merge remote-tracking branch 'origin/main' into main

parents 7933d6fd 56687868
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