文件名称:知识蒸馏KnowledgeDistillation.zip
文件大小:38.07MB
文件格式:ZIP
更新时间:2024-03-30 11:27:05
神经网络 模型压缩
本资源是知识蒸馏的相关代码:老师模型+学生模型,以及学生模型需要用到的KDLoss
【文件预览】:
KnowledgeDistillation
----student_kd.pt(430KB)
----studentModel.py(813B)
----teacher.pt(4.58MB)
----contrastThem.py(1KB)
----test.py(753B)
----teacherModel.py(1KB)
----__pycache__()
--------studentModel.cpython-37.pyc(1KB)
--------teacherModel.cpython-37.pyc(2KB)
--------teacher.cpython-37.pyc(3KB)
--------studentDK.cpython-37.pyc(3KB)
--------KDloss.cpython-37.pyc(812B)
--------student.cpython-37.pyc(3KB)
----student.pt(430KB)
----DarkKnowledge.py(2KB)
----.idea()
--------misc.xml(185B)
--------KnowledgeDistillation.iml(284B)
--------modules.xml(294B)
--------workspace.xml(12KB)
--------.gitignore(176B)
--------inspectionProfiles()
----mnist_data()
--------.DS_Store(6KB)
--------MNIST()
----teacher.py(4KB)
----studentDK.py(4KB)
----student.py(3KB)
----KDloss.py(859B)