KhangTruong commited on
Commit
0592dfd
·
verified ·
1 Parent(s): a64b05f

checkpoint: update KhangTruong/diffusion-diff-minimized-512 (periodic at epoch 3, step 3643)

Browse files
checkpoints/checkpoint_latest.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e31721b8d4d4924f93783aa69ba2cf15e0509003b43ecc1f52ffda1840bb5ed8
3
  size 207855059
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:51e556e7f70ca958639cc30e28ebb9fbed47aa98b1d9fc0878b0132938df70c9
3
  size 207855059
checkpoints/checkpoint_periodic.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1c0d17075c77b13de26aadb02f935799b336ea7bcd31bceee67a46eefe303fc
3
+ size 207856373
checkpoints/checkpoint_periodic_config.yaml ADDED
@@ -0,0 +1,157 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ project:
2
+ name: diffusion_diff_minimized_512
3
+ seed: 42
4
+ device: auto
5
+ output_dir: outputs/RUN/diffusion_diff_minimized_512
6
+ hub_repo: null
7
+ resume_repo: null
8
+ data:
9
+ dataset_name: KhangTruong/DEAL300K-30K
10
+ streaming: true
11
+ image_size:
12
+ - 512
13
+ - 512
14
+ batch_size: 16
15
+ num_workers: -1
16
+ pin_memory: true
17
+ shuffle_buffer_size: 32
18
+ train_split: train
19
+ val_split: validation
20
+ test_split: validation
21
+ eval_split: validation
22
+ train_samples_per_epoch: -1
23
+ val_samples_per_epoch: -1
24
+ val_samples: -1
25
+ test_samples: -1
26
+ evaluate_on_test: true
27
+ augmentations:
28
+ horizontal_flip: 0.5
29
+ vertical_flip: 0.0
30
+ random_rotate90: 0.25
31
+ color_jitter: 0.0
32
+ prefetch_batches: 48
33
+ save_to_disk: true
34
+ cached_hf_repo: KhangTruong/DEAL300K-30K-cache
35
+ base_batch_size: 64
36
+ num_gpus: 1
37
+ _batch_size_scaled: true
38
+ model:
39
+ name: diffusion_diff_minimized
40
+ in_channels: 3
41
+ out_channels: 1
42
+ features:
43
+ - 64
44
+ - 128
45
+ - 256
46
+ - 512
47
+ bilinear: true
48
+ dropout: 0.1
49
+ aux_classifier: true
50
+ num_classes: 3
51
+ gradient_checkpointing: false
52
+ pretrained_model_name_or_path: segmind/tiny-sd
53
+ vae_subfolder: vae
54
+ unet_subfolder: unet
55
+ diffuser_fp16: true
56
+ device_map: cuda
57
+ freeze_encoder: true
58
+ use_encoder_skips: false
59
+ use_perpendicular_skips: true
60
+ timesteps:
61
+ - 250
62
+ timestep_embed_dim: 32
63
+ sigma_embed_dim: 32
64
+ include_noisy_latents: true
65
+ include_added_noise: true
66
+ include_predicted_noise: true
67
+ include_noise_diff: true
68
+ include_z0: true
69
+ scaling_factor: 0.18215
70
+ decoder:
71
+ channels:
72
+ - 256
73
+ - 128
74
+ - 64
75
+ - 32
76
+ upsample_mode: bilinear
77
+ norm_layer: batchnorm
78
+ z_norm: groupnorm
79
+ activation: silu
80
+ dropout: 0.1
81
+ num_res_blocks: 1
82
+ input_rescale: true
83
+ loss:
84
+ mask_loss_type: combined
85
+ bce_weight: 0.5
86
+ dice_weight: 0.5
87
+ focal_weight: 0.5
88
+ focal_gamma: 2.0
89
+ focal_alpha: 0.75
90
+ aux_loss_type: cross_entropy
91
+ aux_weight: 0.2
92
+ pos_weight: 1.0
93
+ aux_classifier: true
94
+ training:
95
+ epochs: 10
96
+ learning_rate: 0.0003
97
+ weight_decay: 0.0001
98
+ optimizer: adamw
99
+ use_8bit_optimizer: false
100
+ check_8bit_compatibility: true
101
+ fallback_to_16bit: true
102
+ fallback_on_unsupported_8bit: true
103
+ scheduler: cosine
104
+ resume_lr_mode: auto
105
+ resume_lr: null
106
+ warmup_epochs: 1
107
+ min_lr: 1.0e-06
108
+ grad_clip_norm: 1.0
109
+ amp: true
110
+ gradient_accumulation_steps: 16
111
+ data_parallel: false
112
+ use_hard_mining: true
113
+ auto_batch_size: true
114
+ empty_cache_per_epoch: true
115
+ save_best: true
116
+ save_latest: true
117
+ checkpoint_period: 3600
118
+ eval_interval: 1
119
+ early_stopping_patience: 4
120
+ early_stopping_metric: val_iou
121
+ early_stopping_mode: max
122
+ push_to_hub: true
123
+ hub_repo: KhangTruong/diffusion-diff-minimized-512
124
+ hub_version: v1
125
+ hf_repo: KhangTruong/diffusion-diff-minimized-512
126
+ num_gpus: 1
127
+ hard_mining:
128
+ enabled: true
129
+ metric: median
130
+ reset_epochs: 5
131
+ bootstrapping:
132
+ enabled: false
133
+ run_bootstrap: false
134
+ epochs: 30
135
+ num_samples: 512
136
+ batch_size: null
137
+ learning_rate: 0.0005
138
+ freeze_strategy: channel_stream
139
+ stream_ratio: 0.5
140
+ release_mode: restore
141
+ freeze_modules: []
142
+ initialization: kaiming_normal
143
+ target_score: 0.5
144
+ min_loss_drop: 0.15
145
+ early_stopping: false
146
+ patience: 3
147
+ logging:
148
+ log_interval: 20
149
+ log_memory: true
150
+ measure_network: true
151
+ post_processing:
152
+ enabled: true
153
+ min_area: 64
154
+ fill_holes: true
155
+ max_hole_size: 256
156
+ morphology: open_close
157
+ morph_kernel_size: 3
manifest.json CHANGED
@@ -9,10 +9,11 @@
9
  "best_epoch": "2",
10
  "total_epochs": "2",
11
  "step": "2813",
12
- "updated_at": "2026-11-02T07:46:30.771857+00:00",
13
  "files": [
14
  "checkpoint_best.pt",
15
- "checkpoint_latest.pt"
 
16
  ]
17
  }
18
  },
 
9
  "best_epoch": "2",
10
  "total_epochs": "2",
11
  "step": "2813",
12
+ "updated_at": "2026-11-02T07:58:53.430514+00:00",
13
  "files": [
14
  "checkpoint_best.pt",
15
+ "checkpoint_latest.pt",
16
+ "checkpoint_periodic.pt"
17
  ]
18
  }
19
  },
versions/v1/checkpoint_periodic_config.yaml ADDED
@@ -0,0 +1,157 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ project:
2
+ name: diffusion_diff_minimized_512
3
+ seed: 42
4
+ device: auto
5
+ output_dir: outputs/RUN/diffusion_diff_minimized_512
6
+ hub_repo: null
7
+ resume_repo: null
8
+ data:
9
+ dataset_name: KhangTruong/DEAL300K-30K
10
+ streaming: true
11
+ image_size:
12
+ - 512
13
+ - 512
14
+ batch_size: 16
15
+ num_workers: -1
16
+ pin_memory: true
17
+ shuffle_buffer_size: 32
18
+ train_split: train
19
+ val_split: validation
20
+ test_split: validation
21
+ eval_split: validation
22
+ train_samples_per_epoch: -1
23
+ val_samples_per_epoch: -1
24
+ val_samples: -1
25
+ test_samples: -1
26
+ evaluate_on_test: true
27
+ augmentations:
28
+ horizontal_flip: 0.5
29
+ vertical_flip: 0.0
30
+ random_rotate90: 0.25
31
+ color_jitter: 0.0
32
+ prefetch_batches: 48
33
+ save_to_disk: true
34
+ cached_hf_repo: KhangTruong/DEAL300K-30K-cache
35
+ base_batch_size: 64
36
+ num_gpus: 1
37
+ _batch_size_scaled: true
38
+ model:
39
+ name: diffusion_diff_minimized
40
+ in_channels: 3
41
+ out_channels: 1
42
+ features:
43
+ - 64
44
+ - 128
45
+ - 256
46
+ - 512
47
+ bilinear: true
48
+ dropout: 0.1
49
+ aux_classifier: true
50
+ num_classes: 3
51
+ gradient_checkpointing: false
52
+ pretrained_model_name_or_path: segmind/tiny-sd
53
+ vae_subfolder: vae
54
+ unet_subfolder: unet
55
+ diffuser_fp16: true
56
+ device_map: cuda
57
+ freeze_encoder: true
58
+ use_encoder_skips: false
59
+ use_perpendicular_skips: true
60
+ timesteps:
61
+ - 250
62
+ timestep_embed_dim: 32
63
+ sigma_embed_dim: 32
64
+ include_noisy_latents: true
65
+ include_added_noise: true
66
+ include_predicted_noise: true
67
+ include_noise_diff: true
68
+ include_z0: true
69
+ scaling_factor: 0.18215
70
+ decoder:
71
+ channels:
72
+ - 256
73
+ - 128
74
+ - 64
75
+ - 32
76
+ upsample_mode: bilinear
77
+ norm_layer: batchnorm
78
+ z_norm: groupnorm
79
+ activation: silu
80
+ dropout: 0.1
81
+ num_res_blocks: 1
82
+ input_rescale: true
83
+ loss:
84
+ mask_loss_type: combined
85
+ bce_weight: 0.5
86
+ dice_weight: 0.5
87
+ focal_weight: 0.5
88
+ focal_gamma: 2.0
89
+ focal_alpha: 0.75
90
+ aux_loss_type: cross_entropy
91
+ aux_weight: 0.2
92
+ pos_weight: 1.0
93
+ aux_classifier: true
94
+ training:
95
+ epochs: 10
96
+ learning_rate: 0.0003
97
+ weight_decay: 0.0001
98
+ optimizer: adamw
99
+ use_8bit_optimizer: false
100
+ check_8bit_compatibility: true
101
+ fallback_to_16bit: true
102
+ fallback_on_unsupported_8bit: true
103
+ scheduler: cosine
104
+ resume_lr_mode: auto
105
+ resume_lr: null
106
+ warmup_epochs: 1
107
+ min_lr: 1.0e-06
108
+ grad_clip_norm: 1.0
109
+ amp: true
110
+ gradient_accumulation_steps: 16
111
+ data_parallel: false
112
+ use_hard_mining: true
113
+ auto_batch_size: true
114
+ empty_cache_per_epoch: true
115
+ save_best: true
116
+ save_latest: true
117
+ checkpoint_period: 3600
118
+ eval_interval: 1
119
+ early_stopping_patience: 4
120
+ early_stopping_metric: val_iou
121
+ early_stopping_mode: max
122
+ push_to_hub: true
123
+ hub_repo: KhangTruong/diffusion-diff-minimized-512
124
+ hub_version: v1
125
+ hf_repo: KhangTruong/diffusion-diff-minimized-512
126
+ num_gpus: 1
127
+ hard_mining:
128
+ enabled: true
129
+ metric: median
130
+ reset_epochs: 5
131
+ bootstrapping:
132
+ enabled: false
133
+ run_bootstrap: false
134
+ epochs: 30
135
+ num_samples: 512
136
+ batch_size: null
137
+ learning_rate: 0.0005
138
+ freeze_strategy: channel_stream
139
+ stream_ratio: 0.5
140
+ release_mode: restore
141
+ freeze_modules: []
142
+ initialization: kaiming_normal
143
+ target_score: 0.5
144
+ min_loss_drop: 0.15
145
+ early_stopping: false
146
+ patience: 3
147
+ logging:
148
+ log_interval: 20
149
+ log_memory: true
150
+ measure_network: true
151
+ post_processing:
152
+ enabled: true
153
+ min_area: 64
154
+ fill_holes: true
155
+ max_hole_size: 256
156
+ morphology: open_close
157
+ morph_kernel_size: 3
versions/v1/checkpoints/checkpoint_latest.pt CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e31721b8d4d4924f93783aa69ba2cf15e0509003b43ecc1f52ffda1840bb5ed8
3
  size 207855059
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:51e556e7f70ca958639cc30e28ebb9fbed47aa98b1d9fc0878b0132938df70c9
3
  size 207855059
versions/v1/checkpoints/checkpoint_periodic.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1c0d17075c77b13de26aadb02f935799b336ea7bcd31bceee67a46eefe303fc
3
+ size 207856373
versions/v1/manifest.json CHANGED
@@ -9,10 +9,11 @@
9
  "best_epoch": "2",
10
  "total_epochs": "2",
11
  "step": "2813",
12
- "updated_at": "2026-11-02T07:46:30.771857+00:00",
13
  "files": [
14
  "checkpoint_best.pt",
15
- "checkpoint_latest.pt"
 
16
  ]
17
  }
18
  },
 
9
  "best_epoch": "2",
10
  "total_epochs": "2",
11
  "step": "2813",
12
+ "updated_at": "2026-11-02T07:58:53.430514+00:00",
13
  "files": [
14
  "checkpoint_best.pt",
15
+ "checkpoint_latest.pt",
16
+ "checkpoint_periodic.pt"
17
  ]
18
  }
19
  },