rlightning.models.toy_model.cnn

class rlightning.models.toy_model.cnn.NatureCNN(sample_obs)

Bases: torch.nn.Module

a simple CNN model for test purpose

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