rlightning.models.vision.cnn

class rlightning.models.vision.cnn.NatureCNN(standard_input_size: Tuple[int, int] = (84, 84), in_channels: int = 3, image_format: str = 'CHW')

Bases: torch.nn.Module

Construct a CNN with given dummy RGB

forward(obs: MockModule('torch.Tensor') | MockModule('numpy.ndarray'))