'GeneratorFullModel' object has no attribute 'mb'

[baihei-1122]

Hi, it's a very nice work.
But when I train the model, I got a problem with 'GeneratorFullModel' object has no attribute 'mb', the 'train.py' shows in 163 lines, I don't find any code about 'mb' define.

[harlanhong]

OK, let me check.

[Utkarsh-shift]

Same issue

[harlanhong]

I will check.