Fix up RmsProp trainer!