def set_gradient_clipping(self, img_clip_value, cap_clip_value):
self.grad_clipping = True
self.img_clipper = gradient_clipping(img_clip_value)
self.cap_clipper = gradient_clipping(cap_clip_value)
self.img_clipper.register_hook(self.img_embedder)
self.cap_clipper.register_hook(self.cap_embedder)
def set_gradient_clipping(self, tf_value):
self.grad_clipping = True
self.tf_clipper = gradient_clipping(tf_value)
self.tf_clipper.register_hook(self.encoder)