Dropout at test time in densenet

6

You can do something like this:
Suppose model has dropout layers.

def apply_dropout(m):
    if type(m) == nn.Dropout:
        m.train()

model.eval()

model.apply(apply_dropout)
6 Likes