Skip to content

triplet_forecasting

defaults:
  - input_encoder: triplet_encoder
  - backbone: triplet_transformer_decoder
  - optimizer: adam
  - scheduler: null
  - _self_
_target_: meds_torch.models.triplet_forecasting.TripletForecastingModule.initialize

max_seq_len: ${data.max_seq_len}
token_dim: 4
vocab_size: ${data.vocab_size}
get_representations: false
task_name: ${data.task_name}
batch_size: ${data.dataloader.batch_size}

# compile model for faster training with pytorch 2.0
compile: false