set quantum device batch size

mit-han-lab · Nov 18, 2024 · f62d372 · f62d372
1 parent 611cc2a
commit f62d372
Showing 1 changed file with 1 addition 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)