multi-source-domain.yaml

experiment:
  name: fine-tuned-seqsleepnet
  module: physioex.train.multi_source
  class: FineTunedModule
  module_config: # this is the argument passed to your model constructor
    n_classes: 5
    model_params:
      name: seqsleepnet
      in_channels: 1
      val_dataloader: null
    seq_len: 21
    loss_call: "cel"
    loss_params: null    
  input_transform: xsleepnet
  target_transform: null


checkpoint: "models/multi-source-domain/"
val_check_interval: 3
max_epoch: 1
batch_size: 128