set quantum device batch size

ankhoa1212 · ankhoa1212 · commit f62d37273f83 · 2024-11-18T03:09:31.000-08:00
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)

Original file line number	Diff line number	Diff line change
`@@ -60,6 +60,7 @@ def forward(self, x, use_qiskit=False):`
`60`	`60`	`self.q_device, self.encoder, self.q_layer, self.measure, x`
`61`	`61`	`)`
`62`	`62`	`else:`
	`63`	`+ self.q_device = tq.QuantumDevice(n_wires=self.n_wires, bsz=bsz)`
`63`	`64`	`self.encoder(self.q_device, x)`
`64`	`65`	`self.q_layer(self.q_device)`
`65`	`66`	`x = self.measure(self.q_device)`