Hi,

I have an spark dataframe with two columns: image, with the binary content, label with string content.

I would like to know how to train an image classification model using Horovod Spark TorchEstimator.

I guess that the important point is setting transformation_fn parameter of TorchEstimator correctly, but I am having issues.

Best regards,
David