fynance.models.rolling.RollMultiLayerPerceptron.cuda

RollMultiLayerPerceptron.cuda(device=None)

Moves all model parameters and buffers to the GPU.

This also makes associated parameters and buffers different objects. So it should be called before constructing optimizer if the module will live on GPU while being optimized.

Arguments:

device (int, optional): if specified, all parameters will be

copied to that device

Returns:

Module: self