docstr: Calls `self.train_fn`
code: def fit(self, train_data: DataLoader, **kwargs):
      r"""
      Calls `self.train_fn`
      """
      return self.train_fn(
          self,
          dataloader=train_data,
          construct_kwargs=self.construct_kwargs,
          **kwargs,
      )
  
imports: from typing import Callable, cast, List, Optional
  import torch.nn
  from captum._utils.models.model import Model
  from torch import Tensor
  from torch.utils.data import DataLoader
label_desc: Use image node_id 3 for calling the LinearModel obj's underlying member method code with example usage: obj.fit(train_data) and returns: self