update

xiaoyufenfei · May 10, 2020 · 0f0c32e · 0f0c32e
1 parent e4943ed
commit 0f0c32e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -1,4 +1,4 @@
-import os
+import os,sys
 import time
 import torch
 from torch import optim
@@ -22,6 +22,8 @@
 from utils.scheduler.lr_scheduler import WarmupPolyLR
 
 
+sys.setrecursionlimit(1000000)  # solve problem 'maximum recursion depth exceeded'
+
 torch_ver = torch.__version__[:3]
 if torch_ver == '0.3':
     from torch.autograd import Variable