diff --git a/solver.py b/solver.py index b5c209b86d82e6d6a36d41cdf05f5c06e7b86bb6..e6bb64e0deba32083237bf96eec0d0758b8e0a24 100644 --- a/solver.py +++ b/solver.py @@ -107,6 +107,9 @@ class Solver(object): self.global_iter, recon_loss.data[0], self.beta*kld.data[0])) curve_data = [] + if self.global_iter%100000 == 0: + self.traverse() + if self.global_iter >= self.max_iter: out = True break