mxfusion.inference.dist_grad_loop

Members

class mxfusion.inference.dist_grad_loop.DistributedGradLoop

Bases: mxfusion.inference.grad_loop.GradLoop

The class for the main loop for distributed batch/minibatch loop gradient-based optimization.

Parameters:data ([mxnet.ndarray]) – a list of observed variables from the distributed batch/minibatch loop.

split_data(data)