I’m not sure how the posted code is used, but would recommend to explicitly set train() on the dropout module via:
model.eval() # sets all layers to eval
model.drop_layer.train() # resets dropout to train