diff --git a/README.md b/README.md
index 46597f1..4fe893b 100644
--- 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
index 936753f..dd3995a 100644
--- 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