(self, pruned_steps)
| 202 | return num_parameters |
| 203 | |
| 204 | def get_target_sparsity(self, pruned_steps): |
| 205 | target_sparsity = (self.target_sparsity - self.start_sparsity) * \ |
| 206 | min(1, pruned_steps / self.lagrangian_warmup) + self.start_sparsity |
| 207 | return target_sparsity |
| 208 | |
| 209 | def lagrangian_regularization(self, pruned_steps): |
| 210 | target_sparsity = self.get_target_sparsity( |
no outgoing calls
no test coverage detected