analogvnn.nn.module.FullSequential#

Module Contents#

Classes#

FullSequential

A sequential model where backward graph is the reverse of forward graph.
class analogvnn.nn.module.FullSequential.FullSequential(tensorboard_log_dir=None, device=is_cpu_cuda.device)[source]#

Bases: analogvnn.nn.module.Sequential.Sequential

A sequential model where backward graph is the reverse of forward graph.

compile(device: Optional[torch.device] = None, layer_data: bool = True)[source]#

Compile the model and add forward and backward graph.

Parameters:

  • device (torch.device) – The device to run the model on.

  • layer_data (bool) – True if the data of the layers should be compiled.

Returns:

self

Return type:

FullSequential