PyTorch Reminders to Self

 

Using

Let’s say that

# %%
drop = nn.Dropout()
x = torch.ones(1,10)
# %%
drop.train()
drop(x)

# %%
drop.eval()
drop(x)