fix: Restore original complexes

qiboteam · Mar 14, 2024 · c5ba8e7 · c5ba8e7
1 parent 2d78cf4
commit c5ba8e7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/qibo/models/qgan.py b/src/qibo/models/qgan.py
@@ -259,11 +259,11 @@ def generate_real_samples(samples, distribution, real_samples):
         # determine half the size of one batch, for updating the discriminator
         half_samples = int(self.batch_samples / 2)
         if self.initial_params is not None:
-            initial_params = tf.Variable(self.initial_params, dtype=tf.complex64)
+            initial_params = tf.Variable(self.initial_params, dtype=tf.complex128)
         else:
             n = 10 * self.layers * self.nqubits + 2 * self.nqubits
             initial_params = tf.Variable(
-                np.random.uniform(-0.15, 0.15, n), dtype=tf.complex64
+                np.random.uniform(-0.15, 0.15, n), dtype=tf.complex128
             )
 
         optimizer = tf.optimizers.Adadelta(  # pylint: disable=no-member