tensorpack.graph_builder package¶
These are some useful functions if you need to write your own trainers. Otherwise you probably don’t need to use them.
-
class
tensorpack.graph_builder.
DataParallelBuilder
(towers)[source]¶ Bases:
tensorpack.graph_builder.training.GraphBuilder
-
class
tensorpack.graph_builder.
LeastLoadedDeviceSetter
(worker_device, ps_devices)[source]¶ Bases:
object
Helper class to assign variables on the least loaded ps-device.
Usage:
with tf.device(LeastLoadedDeviceSetter(...)): ...