Merge pull request #100 from caitaozhan/master

fix a minor bug in quantumnat.py
mit-han-lab · Mar 29, 2023 · ac8e551 · ac8e551
2 parents f754a1b + c999f94
commit ac8e551
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/quantumnat/quantumnat.py b/examples/quantumnat/quantumnat.py
@@ -84,6 +84,7 @@ def __init__(self):
         self.measure = tq.MeasureAll(tq.PauliZ)
 
     def forward(self, x, use_qiskit=False):
+        self.q_device.reset_states(x.shape[0])
         bsz = x.shape[0]
         x = F.avg_pool2d(x, 6).view(bsz, 16)