Solved by BlackBird123 in post #3 
              
              
                Thanks for your response! I really appreciate it. 
If I’m understanding you correctly, it should be something like this: 
self.mod = nn.Sequential(
            nn.Linear(input_dim,100),
            nn.ReLU(),
            nn.Dropout(0.5),
            nn.Linear(100, 50),
            nn.ReLU(),
       …