Hi @rocketstar31 , could you give me some training and validation sample (with the sample weight). Or the shape and datatype info of the data (and the sample weight)? So, I can generate some fake data by myself to do a runtime debug.