diff options
| author | junyanz <junyanz@berkeley.edu> | 2017-10-06 11:18:52 -0700 |
|---|---|---|
| committer | junyanz <junyanz@berkeley.edu> | 2017-10-06 11:18:52 -0700 |
| commit | 1a764f00f8fced8696592bee776d2154b1e8e5a1 (patch) | |
| tree | d2d404e613f6eca258b26edb6518fe5269a8a0e6 /models/base_model.py | |
| parent | 7800d516596f1a25986b458cddf8b8785bcc7df8 (diff) | |
fix minor bugs
Diffstat (limited to 'models/base_model.py')
| -rw-r--r-- | models/base_model.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/models/base_model.py b/models/base_model.py index 55da1ca..446a903 100644 --- a/models/base_model.py +++ b/models/base_model.py @@ -51,8 +51,8 @@ class BaseModel(): save_filename = '%s_net_%s.pth' % (epoch_label, network_label) save_path = os.path.join(self.save_dir, save_filename) network.load_state_dict(torch.load(save_path)) - - def update_learning_rate(): + # update learning rate (called once every epoch) + def update_learning_rate(self): for scheduler in self.schedulers: scheduler.step() lr = self.optimizers[0].param_groups[0]['lr'] |