search

salesforce / PCL

PyTorch code for "Prototypical Contrastive Learning of Unsupervised Representations"
MIT License
570 stars 83 forks source link

The training time of PCL? #2

Closed zzw-zwzhang closed 4 years ago

zzw-zwzhang commented 4 years ago

Hi, I want to know the training time of PCL on each data set, ImageNet & VOC. And the type of graphics card. Thanks.

LiJunnan1992 commented 4 years ago

Hi, training PCL for 200 epochs on ImageNet takes around 1 week with 4 V100 GPU.

zzw-zwzhang commented 4 years ago

Hi, training PCL for 200 epochs on ImageNet takes around 1 week with 4 V100 GPU.

okey, thanks.