run_name: 'SE3-cond-CA'

logdir: '/path/to/logdir'

wandb_params:

  mode: 'online'  # disabled, offline, online

  entity: 'my_username'

  group: 'bindingmoad'

dataset: 'bindingmoad'

datadir: '/path/to/processed_noH_ca/'

enable_progress_bar: False

num_sanity_val_steps: 0

mode: 'pocket_conditioning'

pocket_representation: 'CA'

virtual_nodes: False

batch_size: 64

lr: 5.0e-4

n_epochs: 1000

num_workers: 2

gpus: 1

clip_grad: True

augment_rotation: False

augment_noise: 0

auxiliary_loss: False

loss_params:

  max_weight: 0.001

  schedule: 'linear'

  clamp_lj: 3.0

egnn_params:

  device: 'cuda'

  edge_cutoff_ligand: null

  edge_cutoff_pocket: 8.0

  edge_cutoff_interaction: 8.0

  reflection_equivariant: False

  edge_embedding_dim: null

  joint_nf: 32

  hidden_nf: 128

  n_layers: 5

  attention: True

  tanh: True

  norm_constant: 1

  inv_sublayers: 1

  sin_embedding: False

  aggregation_method: 'sum'

  normalization_factor: 100  # used if aggregation_method='sum'

diffusion_params:

  diffusion_steps: 500

  diffusion_noise_schedule: 'polynomial_2'  # learned, cosine

  diffusion_noise_precision: 1.0e-5

  diffusion_loss_type: 'l2'  # vlb, l2

  normalize_factors: [1, 4]  # [x, h]

eval_epochs: 25

visualize_sample_epoch: 25

visualize_chain_epoch: 25

eval_params:

  n_eval_samples: 100

  smiles_file: '/path/to/train_smiles.npy'

  n_visualize_samples: 5

  keep_frames: 100