Skip to content
This repository was archived by the owner on Jun 4, 2025. It is now read-only.

Update SparseML Integration to V6.1 #26

Merged
bfineran merged 17 commits into from
Apr 8, 2022
Merged

Update SparseML Integration to V6.1 #26

Changes from 1 commit
Commits
Show all changes
17 commits
Select commit Hold shift + click to select a range
8eaecd2
SparseML integration
KSGulin Mar 25, 2022
1f73552
Add SparseML dependancy
KSGulin Mar 25, 2022
4db2a15
Update: add missing files
KSGulin Mar 26, 2022
70fb4cd
Update requirements.txt
KSGulin Mar 30, 2022
bf225ec
Update: sparseml-nightly support
KSGulin Mar 30, 2022
4b4b932
Merge branch 'update-6.1' of github.com:neuralmagic/yolov5 into updat…
KSGulin Mar 30, 2022
6d3667a
Update: remove model versioning
KSGulin Mar 30, 2022
28579c8
Partial update for multi-stage recipes
KSGulin Apr 5, 2022
e5999d5
Update: multi-stage recipe support
KSGulin Apr 6, 2022
3218a78
Update: remove sparseml dep
KSGulin Apr 6, 2022
bbbcf6b
Fix: multi-stage recipe handeling
KSGulin Apr 6, 2022
140ee49
Fix: multi stage support
KSGulin Apr 7, 2022
912040c
Fix: non-recipe runs
KSGulin Apr 8, 2022
a8dfa0f
Add: legacy hyperparam files
KSGulin Apr 8, 2022
7bdf2e6
Fix: add copy-paste to hyps
KSGulin Apr 8, 2022
20f6f91
Fix: nit
KSGulin Apr 8, 2022
5eadf3a
apply structure fixes
Apr 8, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Update requirements.txt
  • Loading branch information
@KSGulin
KSGulin authored Mar 30, 2022
commit 70fb4cd287950f49b343a6849eed43fe7668e1e4
2 changes: 1 addition & 1 deletion requirements.txt
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -35,4 +35,4 @@ seaborn>=0.11.0
# pycocotools>=2.0 # COCO mAP
# roboflow
thop # FLOPs computation
sparseml[torch,torchvision]>=0.11 # Pruning and Quantization
sparseml[torch,torchvision]>=0.12 # Pruning and Quantization