Merge pull request #289 from ankhoa1212/main

Fix for issue #213
mit-han-lab · Nov 19, 2024 · 6ff80a8 · 6ff80a8
2 parents f6a074b + 3f76570
commit 6ff80a8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/examples/clifford_qnn/mnist_clifford_qnn.py b/examples/clifford_qnn/mnist_clifford_qnn.py
@@ -60,6 +60,7 @@ def forward(self, x, use_qiskit=False):
                 self.q_device, self.encoder, self.q_layer, self.measure, x
             )
         else:
+            self.q_device = tq.QuantumDevice(n_wires=self.n_wires, bsz=bsz)
             self.encoder(self.q_device, x)
             self.q_layer(self.q_device)
             x = self.measure(self.q_device)

diff --git a/torchquantum/operator/op_types.py b/torchquantum/operator/op_types.py
@@ -245,6 +245,7 @@ def forward(
                 params = self.params
 
             if self.clifford_quantization:
+                from torchquantum.util.quantization.clifford_quantization import CliffordQuantizer
                 params = CliffordQuantizer.quantize_sse(params)
             if self.n_wires is None:
                 self.func(q_device, self.wires, params=params, inverse=self.inverse)