1 files changed, 2 insertions, 2 deletions
diff --git a/losses.py b/losses.py
index fd68e199..542aa144 100644
--- a/losses.py
+++ b/losses.py
@@ -17,8 +17,8 @@ class YourModel(tf.keras.Model):
         #perhaps consider cropping to avoid distortion
         self.style_image = transform.resize(style_image, np.shape(style_image), anti_aliasing=True)
         self.x = tf.Variable(tf.random.uniform(np.shape(content_image)), trainable=True) 
-        self.alpha = 1
-        self.beta = 1
+        self.alpha = hp.alpha
+        self.beta = hp.beta
 
         self.optimizer = tf.keras.optimizers.RMSprop(learning_rate=hp.learning_rate, momentum=hp.momentum)