Apply reversible block to inputs.
(self, h, training=True)
| 85 | input_shape[1] // curr_strides[1], filters) |
| 86 | |
| 87 | def call(self, h, training=True): |
| 88 | """Apply reversible block to inputs.""" |
| 89 | |
| 90 | for block in self.blocks: |
| 91 | h = block(h, training=training) |
| 92 | return h |
| 93 | |
| 94 | def backward_grads(self, x, y, dy, training=True): |
| 95 | """Apply reversible block backward to outputs.""" |