class_docstr: Multi-layer perceptron with multi-head classifier
class_docstr_tok: ["Multi-layer","perceptron","with","multi-head","classifier"]
code: def __init__(self, input_size=28 * 28, hidden_size=512):
      super().__init__()
  
      self.features = nn.Sequential(
          nn.Linear(input_size, hidden_size),
          nn.ReLU(inplace=True),
          nn.Dropout(),
      )
      self.classifier = MultiHeadClassifier(hidden_size)
      self._input_size = input_size
  
file_path: avalanche/avalanche/models/simple_mlp.py
imports: import torch.nn
  from avalanche.models.dynamic_modules import MultiTaskModule, MultiHeadClassifier
  from avalanche.models.base_model import BaseModel
label_desc: Use image node_id 1 to create a new MTSimpleMLP object from inherited base classes: MultiTaskModule with example: obj = MTSimpleMLP(input_size, hidden_size)