add pre-trained ppsurf 50 NN model

models/ppsurf_50nn/version_0/checkpoints/last.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c56df17bac4c182d9e2c1759f70836fe38090d6312d7685961c1d5d61aa9b533
+size 165502611

models/ppsurf_50nn/version_0/config.yaml

@@ -0,0 +1,132 @@
+# 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