feat(release): final training report and checkpoints for KhangTruong/Testing-model (v1)
1d7c979 verified | project: | |
| name: test_smoke | |
| seed: 42 | |
| device: cpu | |
| output_dir: /tmp/tmpjx19fv4p/out/RUN/test_smoke | |
| hub_repo: null | |
| resume_repo: null | |
| data: | |
| dataset_name: KhangTruong/IMD2020 | |
| streaming: false | |
| image_size: | |
| - 32 | |
| - 32 | |
| batch_size: 16 | |
| num_workers: 0 | |
| pin_memory: false | |
| shuffle_buffer_size: 1000 | |
| train_split: train | |
| val_split: validation | |
| test_split: test | |
| eval_split: test | |
| train_samples_per_epoch: 2 | |
| val_samples_per_epoch: -1 | |
| val_samples: 2 | |
| test_samples: 2 | |
| evaluate_on_test: true | |
| augmentations: | |
| horizontal_flip: 0.5 | |
| vertical_flip: 0.0 | |
| random_rotate90: 0.0 | |
| color_jitter: 0.0 | |
| base_batch_size: 16 | |
| num_gpus: 1 | |
| _batch_size_scaled: true | |
| model: | |
| name: unet | |
| in_channels: 3 | |
| out_channels: 1 | |
| features: | |
| - 8 | |
| - 16 | |
| bilinear: true | |
| dropout: 0.0 | |
| aux_classifier: false | |
| num_classes: 3 | |
| gradient_checkpointing: false | |
| loss: | |
| mask_loss_type: combined | |
| bce_weight: 0.5 | |
| dice_weight: 0.5 | |
| focal_weight: 0.5 | |
| focal_gamma: 2.0 | |
| focal_alpha: 0.25 | |
| aux_loss_type: cross_entropy | |
| aux_weight: 0.2 | |
| training: | |
| epochs: 1 | |
| learning_rate: 0.001 | |
| weight_decay: 0.0001 | |
| optimizer: adamw | |
| use_8bit_optimizer: false | |
| check_8bit_compatibility: true | |
| fallback_to_16bit: true | |
| fallback_on_unsupported_8bit: true | |
| scheduler: cosine | |
| resume_lr_mode: auto | |
| resume_lr: null | |
| warmup_epochs: 1 | |
| min_lr: 1.0e-06 | |
| grad_clip_norm: 1.0 | |
| amp: false | |
| gradient_accumulation_steps: 1 | |
| data_parallel: false | |
| use_hard_mining: false | |
| auto_batch_size: true | |
| empty_cache_per_epoch: true | |
| save_best: true | |
| save_latest: true | |
| checkpoint_period: 3600 | |
| eval_interval: 1 | |
| early_stopping_patience: 2 | |
| early_stopping_metric: val_iou | |
| early_stopping_mode: max | |
| push_to_hub: true | |
| hub_repo: KhangTruong/Testing-model | |
| hub_version: v1 | |
| batch_size: 2 | |
| hf_repo: KhangTruong/Testing-model | |
| num_gpus: 1 | |
| hard_mining: | |
| enabled: false | |
| metric: median | |
| reset_epochs: 5 | |
| bootstrapping: | |
| enabled: false | |
| run_bootstrap: false | |
| epochs: 5 | |
| num_samples: 512 | |
| batch_size: null | |
| learning_rate: null | |
| freeze_strategy: channel_stream | |
| stream_ratio: 0.5 | |
| release_mode: restore | |
| freeze_modules: [] | |
| initialization: kaiming_normal | |
| target_score: 0.5 | |
| min_loss_drop: 0.15 | |
| early_stopping: false | |
| patience: 3 | |
| logging: | |
| log_interval: 1 | |
| log_memory: true | |
| measure_network: true | |
| post_processing: | |
| enabled: true | |
| min_area: 64 | |
| fill_holes: true | |
| max_hole_size: 256 | |
| morphology: open_close | |
| morph_kernel_size: 3 | |