文件名称:moco.tensorflow:动量对比度(MoCo)的TensorFlow重新实现
文件大小:15KB
文件格式:ZIP
更新时间:2024-05-12 23:53:43
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)