analogvnn.nn.module.FullSequential
#
Module Contents#
Classes#
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: