From 600fb825689ecafc1f1abce8b0b3bd6fcd27e355 Mon Sep 17 00:00:00 2001 From: Shawn-Shan Date: Sat, 22 May 2021 10:14:36 -0500 Subject: [PATCH] minor fix --- fawkes/differentiator.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/fawkes/differentiator.py b/fawkes/differentiator.py index 5179fc2..682028b 100644 --- a/fawkes/differentiator.py +++ b/fawkes/differentiator.py @@ -191,7 +191,8 @@ class FawkesMaskGeneration: simg_tanh = self.preprocess_arctanh(source_imgs) if target_imgs is not None: timg_tanh = self.preprocess_arctanh(target_imgs) - self.modifier = tf.Variable(np.ones(tuple([len(source_imgs)] + self.single_shape), dtype=np.float32) * 1e-4) + self.modifier = tf.Variable(np.random.uniform(-1, 1, tuple([len(source_imgs)] + self.single_shape)) * 1e-4, + dtype=tf.float32) # make the optimizer optimizer = tf.keras.optimizers.Adadelta(float(self.learning_rate))