WXinlong / DenseCL

Dense Contrastive Learning (DenseCL) for self-supervised representation learning, CVPR 2021 Oral.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Training speed

hzhupku opened this issue · comments

Could you provide me with the training log. My training process is extremely slow. Thank you.

@hzhupku Your machine might have a slow data loading issue, you are suggested to convert ImageNet to lmdb format through folder2lmdb_imagenet.py, and use this config for training.

For COCO dataset, I didn't encounter this issue. But for convenience, the scripts for converting COCO dataset to lmdb format will be uploaded soon.

@hzhupku FYI, the scripts for converting COCO dataset to lmdb format have been added.