configs/task/DTI.yaml

task: binary
num_classes: null

out:
  _target_: torch.nn.LazyLinear
  out_features: 1

loss:
  _target_: torch.nn.BCEWithLogitsLoss

activation:
  _target_: torch.nn.Sigmoid