Как использовать tf.Dataset для загрузки данных в несколько графических процессоров?

Question

В настоящее время я передаю данные в несколько графических процессоров, используя get_next (). Есть ли лучший способ подачи данных в несколько графических процессоров?

Answer 1

Взгляните на метод класса shard , который предназначен для распределенного обучения. Это может быть то, что вам нужно.