# pytorch_lightning==2.0.9
seed_everything: 42
trainer:
  accelerator: gpu
  strategy: ddp
  devices: -1
  num_nodes: 1
  precision: 16-mixed
  logger:
    class_path: pytorch_lightning.loggers.TensorBoardLogger
    init_args:
      save_dir: models
      version: null
      log_graph: false
      default_hp_metric: true
      prefix: ''
      sub_dir: null
  callbacks:
  - class_path: source.cli.PPSProgressBar
    init_args:
      refresh_rate: 1
      process_position: 0
  - class_path: pytorch_lightning.callbacks.LearningRateMonitor
    init_args:
      logging_interval: step
      log_momentum: false
  - class_path: pytorch_lightning.callbacks.ModelCheckpoint
    init_args:
      dirpath: null
      filename: null
      monitor: null
      verbose: false
      save_last: true
      save_top_k: 0
      save_weights_only: false
      mode: min
      auto_insert_metric_name: true
      every_n_train_steps: null
      train_time_interval: null
      every_n_epochs: null
      save_on_train_epoch_end: null
  fast_dev_run: false
  max_epochs: 150
  min_epochs: null
  max_steps: -1
  min_steps: null
  max_time: null
  limit_train_batches: null
  limit_val_batches: null
  limit_test_batches: null
  limit_predict_batches: null
  overfit_batches: 0.0
  val_check_interval: null
  check_val_every_n_epoch: 1
  num_sanity_val_steps: 0
  log_every_n_steps: 1
  enable_checkpointing: null
  enable_progress_bar: null
  enable_model_summary: null
  accumulate_grad_batches: 1
  gradient_clip_val: null
  gradient_clip_algorithm: null
  deterministic: null
  benchmark: null
  inference_mode: true
  use_distributed_sampler: true
  profiler: null
  detect_anomaly: false
  barebones: false
  plugins: null
  sync_batchnorm: false
  reload_dataloaders_every_n_epochs: 0
  default_root_dir: models/ppsurf_50nn
debug: false
ckpt_path: null
data:
  class_path: source.ppsurf_data_loader.PPSurfDataModule
  init_args:
    in_file: datasets/abc_train/testset.txt
    workers: 48
    use_ddp: true
    padding_factor: 0.05
    seed: 42
    manifold_points: 10000
    patches_per_shape: -1
    do_data_augmentation: true
    batch_size: 12
model:
  class_path: source.ppsurf_model.PPSurfModel
  init_args:
    pointnet_latent_size: 256
    output_names:
    - imp_surf_sign
    in_channels: 3
    out_channels: 2
    k: 64
    lambda_l1: 0.0
    debug: false
    results_dir: results
    name: ppsurf_50nn
    network_latent_size: 256
    gen_subsample_manifold_iter: 10
    gen_subsample_manifold: 10000
    gen_resolution_global: 257
    num_pts_local: 50
    rec_batch_size: 50000
    gen_refine_iter: 10
    workers: 48
optimizer:
  class_path: torch.optim.AdamW
  init_args:
    lr: 0.001
    betas:
    - 0.9
    - 0.999
    eps: 1.0e-05
    weight_decay: 0.01
    amsgrad: false
    maximize: false
    foreach: null
    capturable: false
    differentiable: false
    fused: null
lr_scheduler:
  class_path: torch.optim.lr_scheduler.MultiStepLR
  init_args:
    milestones:
    - 75
    - 125
    gamma: 0.1
    last_epoch: -1
    verbose: false