From 72d5b58b9a8ccc42943576f25441035d73968f15 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Thu, 9 Jul 2020 15:16:57 -0700 Subject: [PATCH] disable LR plot to suppress warning message --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 77e090f..39d2392 100644 --- a/train.py +++ b/train.py @@ -108,7 +108,7 @@ def train(hyp): # Scheduler https://arxiv.org/pdf/1812.01187.pdf lf = lambda x: (((1 + math.cos(x * math.pi / epochs)) / 2) ** 1.0) * 0.9 + 0.1 # cosine scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lf) - plot_lr_scheduler(optimizer, scheduler, epochs, save_dir=log_dir) + # plot_lr_scheduler(optimizer, scheduler, epochs, save_dir=log_dir) # Load Model google_utils.attempt_download(weights)