上传者: 42113456
|
上传时间: 2021-10-30 16:49:23
|
文件大小: 15KB
|
文件类型: -
实施并复制以下论文的结果:
依存关系:
具有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