accelerator: cuda:0
benchmark: false
bn:
  eps: 1.0e-05
  mom: 0.1
cfg_dest: config.yaml
cfg_file: src/configs/ocb-101-GRIT-RRWP.yaml
custom_metrics: []
dataset:
  cache_load: false
  cache_save: false
  dir: ./datasets
  directed: false
  edge_dim: 128
  edge_encoder: false
  edge_encoder_bn: false
  edge_encoder_name: TypeDictEdge
  edge_message_ratio: 0.8
  edge_negative_sampling_ratio: 1.0
  edge_train_mode: all
  encoder: true
  encoder_bn: true
  encoder_dim: 128
  encoder_name: db
  format: custom
  label_column: none
  label_table: none
  location: local
  name: ocb_CktBench101
  nedge_types: 2
  nnode_features: 101
  nnode_types: 11
  node_encoder: false
  node_encoder_bn: false
  node_encoder_name: OCBNode
  node_features_dim: 1
  pe_transform_on_the_fly: false
  remove_feature: false
  resample_disjoint: false
  resample_negative: false
  shuffle_split: true
  slic_compactness: 10
  split:
  - 0.8
  - 0.1
  - 0.1
  split_mode: random
  subcircuit: false
  task: graph
  task_type: generative
  to_undirected: false
  transductive: false
  transform: none
  tu_simple: true
  use_pins: false
device: cuda
devices: 1
edge_ratio: 0.12316987434408007
experiment_name: l-dfm_defog_topo_npx4_npe6
framework:
  type: defog
gnn:
  act: relu
  agg: mean
  att_final_linear: false
  att_final_linear_bn: false
  att_heads: 1
  batchnorm: true
  clear_feature: true
  dim_inner: 64
  dropout: 0.0
  head: san_graph
  keep_edge: 0.5
  l2norm: true
  layer_type: generalconv
  layers_mp: 2
  layers_post_mp: 3
  layers_pre_mp: 0
  msg_direction: single
  normalize_adj: false
  self_msg: concat
  skip_every: 1
  stage_type: stack
gpu_mem: false
gt:
  attn:
    O_e: true
    act: relu
    clamp: 5.0
    deg_scaler: true
    edge_enhance: true
    full_attn: true
    fwl: false
    norm_e: true
    sparse: false
    use: true
    use_bias: true
    x_f_coupling: true
  attn_dropout: 0.2
  batch_norm: false
  bigbird:
    add_cross_attention: false
    attention_type: block_sparse
    block_size: 3
    chunk_size_feed_forward: 0
    hidden_act: relu
    is_decoder: false
    layer_norm_eps: 1.0e-06
    max_position_embeddings: 128
    num_random_blocks: 3
    use_bias: false
  bn_momentum: 0.01
  bn_no_runner: false
  conditional_dim: 0
  conditional_gen: false
  dim_hidden: 64
  dropout: 0.0
  full_graph: true
  gamma: 1.0e-05
  guidance_strength: 2
  layer_norm: true
  layer_type: GritTransformer
  layers: 10
  n_heads: 8
  n_rbf_centroids: 100
  node_pruning: 0
  pna_degrees: []
  residual: true
  sample_separate_t: true
  sizing: false
  time_conditioning: true
  update_e: true
loss:
  edge_weight: 3
  feature_weight: 0.05
mem:
  inplace: false
metric_agg: argmax
metric_best: auto
mlflow:
  name: ocb-GRIT-RRWP
  project: Exp
  use: false
model:
  edge_decoding: dot
  graph_pooling: add
  loss_fun: l1
  match_upper: true
  size_average: mean
  thresh: 0.5
  type: GritTransformer
name_tag: l-dfm_defog_topo_npx4_npe6
node_type_pmf:
- 0.10093013197183609
- 0.2192227691411972
- 0.14002123475074768
- 0.14075174927711487
- 0.020998088642954826
- 0.021558716893196106
- 0
- 0
- 0.0849437266588211
- 0.0849437266588211
- 0.18662986159324646
num_threads: 6
num_workers: 0
optim:
  base_lr: 0.001
  batch_accumulation: 1
  clip_grad_norm: true
  early_stop_by_lr: false
  early_stop_by_perf: false
  lr_decay: 0.1
  max_epoch: 600
  min_lr: 1.0e-06
  min_lr_mode: threshold
  momentum: 0.9
  num_cycles: 0.5
  num_warmup_epochs: 50
  optimizer: adamW
  reduce_factor: 0.5
  schedule_patience: 10
  scheduler: cosine_with_warmup
  steps:
  - 30
  - 60
  - 90
  stop_patience: 100
  weight_decay: 1.0e-05
out_dir: results/ocb-101-GRIT-RRWP-l-dfm_defog_topo_npx4_npe6
posenc_ElstaticSE:
  dim_pe: 16
  enable: false
  kernel:
    times: []
    times_func: range(10)
  layers: 3
  model: none
  n_heads: 4
  pass_as_var: false
  post_layers: 0
  raw_norm_type: none
posenc_EquivStableLapPE:
  eigen:
    eigvec_norm: L2
    laplacian_norm: sym
    max_freqs: 10
  enable: false
  raw_norm_type: none
posenc_HKdiagSE:
  dim_pe: 16
  enable: false
  kernel:
    times: []
    times_func: ''
  layers: 3
  model: none
  n_heads: 4
  pass_as_var: false
  post_layers: 0
  raw_norm_type: none
posenc_LapPE:
  dim_pe: 16
  eigen:
    eigvec_norm: L2
    laplacian_norm: sym
    max_freqs: 10
  enable: false
  layers: 3
  model: none
  n_heads: 4
  pass_as_var: false
  post_layers: 0
  raw_norm_type: none
posenc_RRWP:
  add_identity: true
  add_inverse: false
  add_node_attr: false
  dim_pe: 16
  enable: true
  ksteps: 12
  layers: 3
  model: none
  n_heads: 4
  pass_as_var: false
  post_layers: 0
  raw_norm_type: none
  spd: false
  spse: false
posenc_RWSE:
  dim_pe: 16
  enable: false
  kernel:
    times: []
    times_func: ''
  layers: 3
  model: none
  n_heads: 4
  pass_as_var: false
  post_layers: 0
  raw_norm_type: none
posenc_SignNet:
  dim_pe: 16
  eigen:
    eigvec_norm: L2
    laplacian_norm: sym
    max_freqs: 10
  enable: false
  layers: 3
  model: none
  n_heads: 4
  pass_as_var: false
  phi_hidden_dim: 64
  phi_out_dim: 4
  post_layers: 0
  raw_norm_type: none
pretrained:
  dir: ''
  freeze_main: false
  reset_prediction_head: true
print: both
round: 4
run_dir: results
seed: 0
share:
  dim_in: 1
  dim_out: 3
  num_splits: 3
tensorboard_agg: true
tensorboard_each_run: false
train:
  auto_resume: false
  batch_size: 32
  ckpt_best: true
  ckpt_clean: true
  ckpt_period: 100
  distortion_pow_e: 6
  distortion_pow_f: 2
  distortion_pow_n: 4
  enable_ckpt: true
  epoch_resume: -1
  eval_period: 5
  iter_per_epoch: 32
  mode: dfm
  neighbor_sizes:
  - 20
  - 15
  - 10
  - 5
  node_per_graph: 32
  noise_feat_only: false
  prior: marginal
  radius: extend
  ratio_cf_guidance: 0.1
  sample_node: false
  sampler: full_batch
  save_final_model: true
  skip_train_eval: false
  t_sample_distortion_e: pow
  t_sample_distortion_f: pow
  t_sample_distortion_n: pow
  use_hungarian: false
  walk_length: 4
val:
  node_per_graph: 32
  radius: extend
  sample_node: false
  sampler: full_batch
view_emb: false
