checkpoint: <TODO>
set: train
sample_outdir: ./samples
n_samples: 50
sample_with_ground_truth_size: False
device: cuda
seed: 42
sample: True
postprocess: False
evaluate: False
reduce: reduce

# Override training config parameters if necessary
model_args:
  
  virtual_nodes: [0, 10]

  train_params:
    datadir: ./processed_crossdocked
    gnina: gnina
    batch_size: 2

  eval_params:
    n_sampling_steps: 100