From 7800d516596f1a25986b458cddf8b8785bcc7df8 Mon Sep 17 00:00:00 2001 From: junyanz Date: Fri, 6 Oct 2017 10:46:43 -0700 Subject: support nc=1, add new leaerning rate policy and new initialization --- models/base_model.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'models/base_model.py') diff --git a/models/base_model.py b/models/base_model.py index 36ceb43..55da1ca 100644 --- a/models/base_model.py +++ b/models/base_model.py @@ -53,4 +53,7 @@ class BaseModel(): network.load_state_dict(torch.load(save_path)) def update_learning_rate(): - pass + for scheduler in self.schedulers: + scheduler.step() + lr = self.optimizers[0].param_groups[0]['lr'] + print('learning rate = %.7f' % lr) -- cgit v1.2.3-70-g09d2