ich777/Frigate_MemryX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

update YOLO_NAS notebook (#17414)

Browse Source 
Google Colab updated to python 3.11
super-gradients v3.7.1 is not compatible with py3.11 and install fails
super-gradients committed a fix to master branch but did not cut a relase since (acquired by Nvidia in the meantime)
This commit installs super-gradients from master branch
This commit is contained in:
aptalca
2025-03-27 12:33:03 -04:00
committed by GitHub
parent 8978d1ff74
commit ccf20f456a
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
notebooks/YOLO_NAS_Pretrained_Export.ipynb
View File

@@ -8,14 +8,14 @@
},
"outputs": [],
"source": [
"! pip install -q super_gradients==3.7.1"
"! pip install -q git+https://github.com/Deci-AI/super-gradients.git"
]
},
{
"cell_type": "code",
"source": [
"! sed -i 's/sghub.deci.ai/sg-hub-nv.s3.amazonaws.com/' /usr/local/lib/python3.10/dist-packages/super_gradients/training/pretrained_models.py\n",
"! sed -i 's/sghub.deci.ai/sg-hub-nv.s3.amazonaws.com/' /usr/local/lib/python3.10/dist-packages/super_gradients/training/utils/checkpoint_utils.py"
"! sed -i 's/sghub.deci.ai/sg-hub-nv.s3.amazonaws.com/' /usr/local/lib/python3.11/dist-packages/super_gradients/training/pretrained_models.py\n",
"! sed -i 's/sghub.deci.ai/sg-hub-nv.s3.amazonaws.com/' /usr/local/lib/python3.11/dist-packages/super_gradients/training/utils/checkpoint_utils.py"
],
"metadata": {
"id": "NiRCt917KKcL"
@@ -84,4 +84,4 @@
},
"nbformat": 4,
"nbformat_minor": 0
}
}