diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-10-06 10:46:43 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-10-06 10:46:43 -0700 |
| commit | 7800d516596f1a25986b458cddf8b8785bcc7df8 (patch) | |
| tree | 56d57350e7104393f939ec7cc2e07c96840aaa27 /models/base_model.py | |
| parent | e986144cee13a921fd3ad68d564f820e8f7dd3b0 (diff) | |
support nc=1, add new leaerning rate policy and new initialization
Diffstat (limited to 'models/base_model.py')
| -rw-r--r-- | models/base_model.py | 5 |
1 files changed, 4 insertions, 1 deletions
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) |