From 2ed6a8d06e7e8269511b7ef2466911f53e3e0720 Mon Sep 17 00:00:00 2001
From: Duzeyao <330501241@qq.com>
Date: Wed, 13 Nov 2019 15:59:20 +0800
Subject: [PATCH] Update train.py

---
 train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/train.py b/train.py
index 009e436..bbab268 100644
--- a/train.py
+++ b/train.py
@@ -107,6 +107,7 @@ def main():
     min_length = args.min_length
     output_dir = args.output_dir
     tb_writer = SummaryWriter(log_dir=args.writer_dir)
+    assert log_step % gradient_accumulation == 0
 
     if not os.path.exists(output_dir):
         os.mkdir(output_dir)