Ok I think I see your problem. I kept overlooking it but it is so simple. When you defined the model and init function you forgot to add the two underscores on each side like this:
class CNNModel1(nn.Module):  # (nn.Module):

    def __init__(self, fully_layer_1=512, fully_layer_2=256,  drop_rate-0.5):
        super(CNNModel1, self).__init__()

hopefully that will fix it.