File size: 3,054 Bytes
1409977 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 |
# 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
|