[SOLVED] Using ApacheBeam with Tensorflow (X) or pythorch


This Content is from Stack Overflow. Question asked by Bennimi

After watching a couple of talks and reading some blogs about ApacheBeam and its API/SDK, I still can´t get my head around whether it is also capable of integrating with distributed Machine Learning training paradigms (e.g. data-parallel).

The content I found online usually only covers how it can help to build an end-to-end ML pipeline (from pre-processing to serving) but leaves out if the training can be also done in parallel.

So my question is: Can one actually integrate it with, say, Tensorflow´s distributed trainings libs, like tf.distribute.MultiWorkerMirroredStrategy ?


There’s not currently a good way to do this kind of training in Beam because Beam’s model assumes that all worker machines in a given step are independent of each other, and it enforces a directed graph (so there’s not an easy way to do native iterative training either).

This could change eventually, but in the short term Beam is better situated for more "embarassingly parallel" operations like inference, pre/post-processing, and some specific forms of training that fit onto a single machine (e.g. training many per-entity models, some kinds of online training)

This Question was asked in StackOverflow by Bennimi and Answered by Daniel It is licensed under the terms of CC BY-SA 2.5. - CC BY-SA 3.0. - CC BY-SA 4.0.

people found this article helpful. What about you?