diff --git a/train.py b/train.py
index 9c8fed87..0c63a514 100644
--- a/train.py
+++ b/train.py
@@ -147,7 +147,7 @@ def train():
     # for _ in range(epochs):
     #     scheduler.step()
     #     y.append(optimizer.param_groups[0]['lr'])
-    # plt.plot(y, label='LambdaLR')
+    # plt.plot(y, '.-', label='LambdaLR')
     # plt.xlabel('epoch')
     # plt.ylabel('LR')
     # plt.tight_layout()