| 48 | |
| 49 | |
| 50 | class ClipCallback(Callback): |
| 51 | def _setup_graph(self): |
| 52 | vars = tf.trainable_variables() |
| 53 | ops = [] |
| 54 | for v in vars: |
| 55 | n = v.op.name |
| 56 | if not n.startswith('discrim/'): |
| 57 | continue |
| 58 | logger.info("Clip {}".format(n)) |
| 59 | ops.append(tf.assign(v, tf.clip_by_value(v, -0.01, 0.01))) |
| 60 | self._op = tf.group(*ops, name='clip') |
| 61 | |
| 62 | def _trigger_step(self): |
| 63 | self._op.run() |
| 64 | |
| 65 | |
| 66 | if __name__ == '__main__': |
no outgoing calls
no test coverage detected
searching dependent graphs…