Adding more ITs for mtraining

Change-Id: I6b44cf58b31ff4de90648673b2b9d1250ee9d3b3