Does Pytorch support training on android devices ( with java codes)

Wonder if pytorch support training on android devices to perform federated learning in practice

1 Like

At the moment we do not support training in pytorch mobile. We are looking on it. Stay tuned.

Regarding Java code, as long as we can build the C++ API for Android, we can use that from Java with the JavaCPP Presets for PyTorch: