Add hardware checks to notebook_init() (#5919)
* Update notebook * Update notebook * update string * update string * Updates * Updates * Updates * check both ipython and psutil * remove sample_data if is_colab * cleanup * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
1 parent
581dc301a7
commit
7d56d45124
2 files changed
+27
-7
No files matched your search
Vendored
+2
-1
@@ -409,6 +409,7 @@
|
||||
"%cd yolov5\n",
|
||||
"%pip install -qr requirements.txt # install\n",
|
||||
"\n",
|
||||
"import torch\n",
|
||||
"from yolov5 import utils\n",
|
||||
"display = utils.notebook_init() # checks"
|
||||
],
|
||||
@@ -983,7 +984,7 @@
|
||||
"source": [
|
||||
"# Reproduce\n",
|
||||
"for x in 'yolov5s', 'yolov5m', 'yolov5l', 'yolov5x':\n",
|
||||
" !python val.py --weights {x}.pt --data coco.yaml --img 640 --conf 0.25 --iou 0.45 # speed\n",
|
||||
" !python val.py --weights {x}.pt --data coco.yaml --img 640 --task speed # speed\n",
|
||||
" !python val.py --weights {x}.pt --data coco.yaml --img 640 --conf 0.001 --iou 0.65 # mAP"
|
||||
],
|
||||
"execution_count": null,
|
||||
|
||||
Reference in new issue
Block a user