Hi, i have a problem. I noticed the self.n[3, 8, 3] in line 185 of model.py, but the third stage of n is 7 in the paper. Maybe self.n[3, 7, 3].