search

WXinlong / DenseCL

Dense Contrastive Learning (DenseCL) for self-supervised representation learning, CVPR 2021 Oral.
GNU General Public License v3.0
544 stars 70 forks source link

Training speed #2

Closed hzhupku closed 3 years ago

hzhupku commented 3 years ago

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

WXinlong commented 3 years ago

@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.

WXinlong commented 3 years ago

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