I applied transformation during training phase in pytorch then I convert my model to run in tensorflow.