Sequential( torch.nn.Linear(784, 200), torch.nn.Dropout(0.5), # drop 50% of the neuron torch.nn.ReLU(), torch.nn.Linear(200, 200), torch.nn.Dropout(0.5), # drop 50% of the neuron torch torch.nn.ReLU(), torch.nn.Linear(200, 10), ## Clarification - torch.nn.Dropout(p=dropout_prob) tf.nn.dropout(keep_prob) ## Behavior between train and test ## ☀️ ☁️ ☁️ for epoch in range(epochs):