UnboundLocalError: local variable 'alpha_mask' referenced before assignment

[HuizhiXu]

When executing self.model.plot(), an error occurs:
UnboundLocalError: local variable 'alpha_mask' referenced before assignment.

Upon inspection of the code, it indicates that when the necessary conditions are not met, the alpha_mask remains undefined.

[KindXiaoming]

Hi, what is the value of symbol_mask and numerical_mask in your case?

[HuizhiXu]

Hi, we have finally reproduced the error. We run 30 steps and call the plot function after each training step. The error occurs during the process: when the training error suddenly becomes NaN, the next step is to plot, and then the error occurs. At this point, both the values of symbol_mask and numerical_mask are NaN.