pytorch-metric-learning icon indicating copy to clipboard operation
pytorch-metric-learning copied to clipboard

Published 20 hours ago • verified publisher icon KevinMusgrave

Add trainer for MoCo

Open KevinMusgrave opened this issue 5 years ago • 0 comments

It would be nice to have a trainer for MoCo. It would be similar to UnsupervisedEmbeddingsUsingAugmentations but would need to use CrossBatchMemory for the queue. Also, since the queue has to contain only negatives, there needs to be some care taken when creating labels inside each batch.

KevinMusgrave avatar May 03 '20 18:05 KevinMusgrave