Spaces:
Runtime error
Runtime error
| # 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 | |