【文件属性】:
文件名称:moco.tensorflow:动量对比度(MoCo)的TensorFlow重新实现
文件大小:15KB
文件格式:ZIP
更新时间:2021-04-23 06:07:03
Python
实施并复制以下论文的结果:
依存关系:
具有XLA支持的TensorFlow 1.14或1.15
≥0.19 GLOO与NCCL支持内置
TensorFlow
OpenCV
taskset命令(来自util-linux包)
无监督培训:
要在具有8个GPU的计算机上运行MoCo预培训,请使用:
horovodrun -np 8 --output-filename moco.log python main_moco.py --data /path/to/imagenet
添加--v2来训练MoCov2,该MoCov2使用额外的MLP层,额外的扩充和余弦LR时间表。
线性分类:
要使用预训练的特征训练线性分类器,请运行:
./main_lincls.py --load /path/to/pretrained/checkpoint --data /path/to/imagene
【文件预览】:
moco.tensorflow-master
----resnet.py(4KB)
----main_lincls.py(6KB)
----serve-data.py(2KB)
----.github()
--------ISSUE_TEMPLATE.md(387B)
----tox.ini(462B)
----.gitignore(986B)
----eval_knn.py(5KB)
----README.md(2KB)
----main_moco.py(11KB)
----data.py(4KB)