update train example with transforms

[Nic-Ma]

Is your feature request related to a problem? Please describe.
As we already have many new transforms, need to update the example code to use them.
And also need to implement "ChangeToChannelFirst" transform for input data.