zhangyi_1015

thx. actually I find the attribute 'done_beam' was in model.module. when I replace the code by( for _ in model.module.done_beams) it works @jamiechoi1995

@ruotianluo thanks for your code！ I have learned a lot through all the code you wrote！