(self, cur_loss=None)
| 56 | c.unfreeze() |
| 57 | |
| 58 | def update(self, cur_loss=None): |
| 59 | assert self.trainable, "Layer is frozen" |
| 60 | for c in self.components: |
| 61 | c.update(cur_loss) |
| 62 | self.flush_gradients() |
| 63 | |
| 64 | def flush_gradients(self): |
| 65 | assert self.trainable, "Layer is frozen" |
no test coverage detected