Update lr
(self, optimizer, epoch)
| 117 | return params |
| 118 | |
| 119 | def update_lr(self, optimizer, epoch): |
| 120 | """Update lr |
| 121 | """ |
| 122 | if epoch > self.config.train.num_epochs_static_embedding: |
| 123 | for param_group in optimizer.param_groups[:2]: |
| 124 | param_group["lr"] = self.config.optimizer.learning_rate |
| 125 | else: |
| 126 | for param_group in optimizer.param_groups[:2]: |
| 127 | param_group["lr"] = 0 |
| 128 | |
| 129 | def forward(self, batch): |
| 130 | raise NotImplementedError |
nothing calls this directly
no outgoing calls
no test coverage detected