From 2cc459e3e4b2d559b5d8aa757c694db02ccd0e2a Mon Sep 17 00:00:00 2001 From: Jordan Gong Date: Thu, 14 Jul 2022 17:41:49 +0800 Subject: Implement InfoNCE loss --- readme.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'readme.md') diff --git a/readme.md b/readme.md index 29dc613..2d5bb74 100644 --- a/readme.md +++ b/readme.md @@ -4,7 +4,7 @@ - [x] Random color distortion (from SimCLR) - [x] Random Gaussian blur (from SimCLR) - [x] Random multi-crop (from SwAV) - - [ ] InfoNCE loss + - [x] InfoNCE loss - [ ] Momentum encoder - [ ] Selective Kernel (SK) convolution layer (used in SimCLR v2) - [x] LARS optimizer -- cgit v1.2.3