Please make sure most of the code is inherited from the MxNet generator. For instance in the generator class.