code: def a_model_with_dropout():
      return torch.nn.Sequential(
          torch.nn.Linear(10, 5),
          torch.nn.ReLU(),
          torch.nn.Sequential(
              torch.nn.Dropout(p=0.5),
              torch.nn.Linear(5, 5),
              torch.nn.ReLU(),
          ),
          torch.nn.Sequential(
              torch.nn.Dropout(p=0.5),
              torch.nn.Linear(5, 2),
          ),
      ).eval()
  
file_path: baal/tests/bayesian/dropout_test.py
imports: import warnings
  import pytest
  import torch
  import baal.bayesian.dropout
label_desc: Use image node_id 1 for calling a global function with example usage: a_model_with_dropout() and returns: torch