(self)
| 63 | self._set_lr(warm_up_lr_l) |
| 64 | |
| 65 | def get_current_learning_rate(self): |
| 66 | return [param_group['lr'] for param_group in self.optimizers[0].param_groups] |
| 67 | |
| 68 | def get_network_description(self, network): |
| 69 | """Get the string and total parameters of the network""" |