Skip to content

Commit 8fdb36b

Browse files
authored
Merge pull request #19 from kevinykuo/feature/flush
flush print buffer for console
2 parents c31d3ef + 5cc7285 commit 8fdb36b

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

ctgan/synthesizer.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -215,7 +215,8 @@ def fit(self, train_data, discrete_columns=tuple(), epochs=300):
215215
optimizerG.step()
216216

217217
print("Epoch %d, Loss G: %.4f, Loss D: %.4f" %
218-
(i + 1, loss_g.detach().cpu(), loss_d.detach().cpu()))
218+
(i + 1, loss_g.detach().cpu(), loss_d.detach().cpu()),
219+
flush=True)
219220

220221
def sample(self, n):
221222
"""Sample data similar to the training data.

0 commit comments

Comments
 (0)