| seed: 42 | |
| paths: | |
| root_dir: D:/AI/ImageAI | |
| master_dir: D:/AI/ImageAI/MasterPNG | |
| sparse_dir: D:/AI/ImageAI/SparsePNG | |
| splits_dir: D:/AI/ImageAI/splits | |
| runs_dir: D:/AI/ImageAI/runs | |
| checkpoints_dir: D:/AI/ImageAI/checkpoints | |
| outputs_dir: D:/AI/ImageAI/outputs | |
| data: | |
| train_split: train_native_even.txt | |
| val_split: val_native_even.txt | |
| test_split: test_native_even.txt | |
| sample_mode: native_image | |
| patch_size: null | |
| eval_patch_size: null | |
| full_frame_size: null | |
| full_frame_pad_mode: edge | |
| train_batch_size: 1 | |
| val_batch_size: 1 | |
| num_workers: 12 | |
| pin_memory: true | |
| persistent_workers: true | |
| prefetch_factor: 2 | |
| train_derive_sparse_from_gt: false | |
| val_derive_sparse_from_gt: false | |
| train_horizontal_flip: false | |
| train_vertical_flip: false | |
| model: | |
| name: sparse_unet | |
| in_channels: 16 | |
| out_channels: 12 | |
| base_channels: 96 | |
| loss: | |
| missing_weight: 1.0 | |
| known_weight: 0.05 | |
| training: | |
| device: cuda | |
| amp: true | |
| max_steps: 1000000 | |
| warmup_steps: 1000 | |
| validate_every: 1000 | |
| save_every: 1000 | |
| log_every: 50 | |
| learning_rate: 0.0002 | |
| min_learning_rate: 0.000001 | |
| weight_decay: 0.01 | |
| betas: [0.9, 0.99] | |
| grad_clip_norm: 1.0 | |
| run_name: sparse_unet_native_bc96 | |
| resume: null | |
| inference: | |
| tile_size: 512 | |
| tile_overlap: 64 | |