class Predictor[source]

Predictor(in_dim, mid_dim, out_dim, p_drop=0.5) :: Module

The Predictor on top of the encoder network

class SimSiamAE[source]

SimSiamAE(device, args) :: AbsModel

The SimSiam contains the Autoencoder based Architecture and the modified Pretext task