momentum2-teacher:实施动量^ 2老师

时间:2024-06-14 07:30:22
【文件属性】:

文件名称:momentum2-teacher:实施动量^ 2老师

文件大小:32KB

文件格式:ZIP

更新时间:2024-06-14 07:30:22

Python

动量^ 2老师:具有动量统计信息的动量老师用于自主学习 要求 所有实验均使用python3.6完成,torch == 1.5.0; 火炬视觉== 0.6.0 用法 资料准备 在${root_of_your_clone}/data/imagenet_train , ${root_of_your_clone}/data/imagenet_val准备ImageNet数据。 由于我们有一个内部平台(存储)来读取imagenet,因此我没有尝试过本地模式。 您可能需要在momentum_teacher/data/dataset.py进行一些修改以支持本地模式。 训练 训练之前,请确保在您的PYTHONPATH中添加了路径(即${root_of_clone} ),例如 export PYTHONPATH=$PYTHONPATH:${root_of_clone} 要在8-gpu机器上的ImageNet


【文件预览】:
momentum2-teacher-main
----.gitignore(3KB)
----Makefile(353B)
----.pre-commit-config.yaml(682B)
----momentum_teacher()
--------exps()
--------data()
--------layers()
--------models()
--------tools()
--------utils()
----Readme.md(6KB)

网友评论