update README to make the pointer to reference models.

hula-ai · Sep 19, 2021 · 4881c3a · 4881c3a
1 parent 9ff4873
commit 4881c3a
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/README.md b/README.md
@@ -176,7 +176,7 @@ This gives us 83.2% accuracy for ViT-Base with 150-epoch fine-tuning.
 
 ### Model Configs
 
-See the commands listed in [CONFIG.md](https://github.com/facebookresearch/moco-v3/blob/main/CONFIG.md).
+See the commands listed in [CONFIG.md](https://github.com/facebookresearch/moco-v3/blob/main/CONFIG.md) for specific model configs, including our recommended hyper-parameters and pre-trained reference models.
 
 ### Transfer Learning
 

diff --git a/moco/optimizer.py b/moco/optimizer.py
@@ -4,7 +4,6 @@
 # This source code is licensed under the license found in the
 # LICENSE file in the root directory of this source tree.
 
-import math
 import torch