So, now that you know what you’re doing  how would you do it?
Also, I noticed the second dropout is from F (not from nn), how are they different? I see that F.dropout takes ‘training’ argument, but nn.dropout doesn’t. How does nn.dropout knows if we’re doing training or testing?