class Mymodel(nn.Module):
    def __init__(self, 
                 base_encoder, 
                 num_classes=21, 
                 moving_average_decay = 0.99,
                 augment_fn = None,
                 augment_fn2 = None,
                 use_momentum=True):
        super(Mymodel, self).__init__()

        self.m = moving_average_decay

        self.online_encoder = base_encoder
        print(type(base_encoder))  # try to debug

try to debug and show the code how you instantiate the Mymodel