anthonymeo
commited on
Commit
•
c6b353a
1
Parent(s):
87280e3
Upload folder using huggingface_hub
Browse files- .hydra/config.yaml +320 -0
- .hydra/hydra.yaml +160 -0
- .hydra/overrides.yaml +1 -0
- average_model.pth +3 -0
- ckpt_best.pth +3 -0
- ckpt_epoch_100.pth +3 -0
- ckpt_latest.pth +3 -0
- console_Nov21_05_01_06.txt +0 -0
- events.out.tfevents.1732165266.1213f0dc8cb8.11180.0 +3 -0
- experiment_logs_Nov21_05_01_06.txt +0 -0
- logs_Nov21_05_01_06.txt +46 -0
.hydra/config.yaml
ADDED
@@ -0,0 +1,320 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
training_hyperparams:
|
2 |
+
resume: null
|
3 |
+
run_id: null
|
4 |
+
resume_path: null
|
5 |
+
resume_from_remote_sg_logger: false
|
6 |
+
ckpt_name: ckpt_latest.pth
|
7 |
+
lr_mode: CosineLRScheduler
|
8 |
+
lr_schedule_function: null
|
9 |
+
lr_warmup_epochs: 3
|
10 |
+
lr_warmup_steps: 1000
|
11 |
+
lr_cooldown_epochs: 0
|
12 |
+
warmup_initial_lr: 1.0e-06
|
13 |
+
step_lr_update_freq: null
|
14 |
+
cosine_final_lr_ratio: 0.1
|
15 |
+
warmup_mode: LinearEpochLRWarmup
|
16 |
+
lr_updates:
|
17 |
+
_target_: super_gradients.training.utils.utils.empty_list
|
18 |
+
pre_prediction_callback: null
|
19 |
+
optimizer: AdamW
|
20 |
+
optimizer_params:
|
21 |
+
weight_decay: 0.0001
|
22 |
+
load_opt_params: true
|
23 |
+
zero_weight_decay_on_bias_and_bn: true
|
24 |
+
loss: PPYoloELoss
|
25 |
+
criterion_params:
|
26 |
+
use_static_assigner: false
|
27 |
+
num_classes: ${num_classes}
|
28 |
+
reg_max: 16
|
29 |
+
ema: true
|
30 |
+
ema_params:
|
31 |
+
decay: 0.9
|
32 |
+
decay_type: threshold
|
33 |
+
beta: 15
|
34 |
+
train_metrics_list: []
|
35 |
+
valid_metrics_list:
|
36 |
+
- DetectionMetrics_050:
|
37 |
+
score_thres: 0.1
|
38 |
+
top_k_predictions: 300
|
39 |
+
num_cls: ${num_classes}
|
40 |
+
normalize_targets: true
|
41 |
+
post_prediction_callback:
|
42 |
+
_target_: super_gradients.training.models.detection_models.pp_yolo_e.PPYoloEPostPredictionCallback
|
43 |
+
score_threshold: 0.01
|
44 |
+
nms_top_k: 1000
|
45 |
+
max_predictions: 300
|
46 |
+
nms_threshold: 0.7
|
47 |
+
metric_to_watch: mAP@0.50
|
48 |
+
greater_metric_to_watch_is_better: true
|
49 |
+
launch_tensorboard: false
|
50 |
+
tensorboard_port: null
|
51 |
+
tb_files_user_prompt: false
|
52 |
+
save_tensorboard_to_s3: false
|
53 |
+
precise_bn: false
|
54 |
+
precise_bn_batch_size: null
|
55 |
+
sync_bn: true
|
56 |
+
silent_mode: false
|
57 |
+
mixed_precision: true
|
58 |
+
save_ckpt_epoch_list:
|
59 |
+
- 100
|
60 |
+
- 200
|
61 |
+
- 250
|
62 |
+
average_best_models: true
|
63 |
+
dataset_statistics: false
|
64 |
+
batch_accumulate: 1
|
65 |
+
run_validation_freq: 1
|
66 |
+
run_test_freq: 1
|
67 |
+
save_model: true
|
68 |
+
seed: 42
|
69 |
+
phase_callbacks: []
|
70 |
+
log_installed_packages: true
|
71 |
+
clip_grad_norm: null
|
72 |
+
ckpt_best_name: ckpt_best.pth
|
73 |
+
max_train_batches: null
|
74 |
+
max_valid_batches: null
|
75 |
+
sg_logger: base_sg_logger
|
76 |
+
sg_logger_params:
|
77 |
+
tb_files_user_prompt: false
|
78 |
+
launch_tensorboard: false
|
79 |
+
tensorboard_port: null
|
80 |
+
save_checkpoints_remote: false
|
81 |
+
save_tensorboard_remote: false
|
82 |
+
save_logs_remote: false
|
83 |
+
monitor_system: true
|
84 |
+
torch_compile: false
|
85 |
+
torch_compile_loss: false
|
86 |
+
torch_compile_options:
|
87 |
+
mode: reduce-overhead
|
88 |
+
fullgraph: false
|
89 |
+
dynamic: false
|
90 |
+
backend: inductor
|
91 |
+
options: null
|
92 |
+
disable: false
|
93 |
+
finetune: false
|
94 |
+
_convert_: all
|
95 |
+
max_epochs: 100
|
96 |
+
initial_lr: 0.0004
|
97 |
+
dataset_params:
|
98 |
+
data_dir: /workspace/final-final-3
|
99 |
+
dataset_name: ${dataset_name}
|
100 |
+
train_dataset_params:
|
101 |
+
data_dir: ${..data_dir}
|
102 |
+
dataset_name: ${..dataset_name}
|
103 |
+
split: train
|
104 |
+
input_dim:
|
105 |
+
- 640
|
106 |
+
- 640
|
107 |
+
cache_annotations: true
|
108 |
+
ignore_empty_annotations: false
|
109 |
+
transforms:
|
110 |
+
- DetectionMosaic:
|
111 |
+
input_dim: ${dataset_params.train_dataset_params.input_dim}
|
112 |
+
prob: 1.0
|
113 |
+
- DetectionRandomAffine:
|
114 |
+
degrees: 0.0
|
115 |
+
translate: 0.1
|
116 |
+
scales:
|
117 |
+
- 0.5
|
118 |
+
- 1.5
|
119 |
+
shear: 0.0
|
120 |
+
target_size: ${dataset_params.train_dataset_params.input_dim}
|
121 |
+
filter_box_candidates: false
|
122 |
+
wh_thr: 2
|
123 |
+
area_thr: 0.1
|
124 |
+
ar_thr: 20
|
125 |
+
border_value: 128
|
126 |
+
- DetectionHSV:
|
127 |
+
prob: 1.0
|
128 |
+
hgain: 5
|
129 |
+
sgain: 30
|
130 |
+
vgain: 30
|
131 |
+
- DetectionHorizontalFlip:
|
132 |
+
prob: 0.5
|
133 |
+
- DetectionPaddedRescale:
|
134 |
+
input_dim: ${dataset_params.train_dataset_params.input_dim}
|
135 |
+
- DetectionStandardize:
|
136 |
+
max_value: 255.0
|
137 |
+
- DetectionTargetsFormatTransform:
|
138 |
+
input_dim: ${dataset_params.train_dataset_params.input_dim}
|
139 |
+
output_format: LABEL_CXCYWH
|
140 |
+
class_inclusion_list: null
|
141 |
+
max_num_samples: null
|
142 |
+
with_crowd: false
|
143 |
+
verbose: 0
|
144 |
+
train_dataloader_params:
|
145 |
+
shuffle: true
|
146 |
+
batch_size: 12
|
147 |
+
min_samples: 512
|
148 |
+
num_workers: 4
|
149 |
+
drop_last: false
|
150 |
+
pin_memory: true
|
151 |
+
worker_init_fn:
|
152 |
+
_target_: super_gradients.training.utils.utils.load_func
|
153 |
+
dotpath: super_gradients.training.datasets.datasets_utils.worker_init_reset_seed
|
154 |
+
collate_fn: DetectionCollateFN
|
155 |
+
val_dataset_params:
|
156 |
+
data_dir: ${..data_dir}
|
157 |
+
dataset_name: ${..dataset_name}
|
158 |
+
split: valid
|
159 |
+
input_dim:
|
160 |
+
- 640
|
161 |
+
- 640
|
162 |
+
cache_annotations: true
|
163 |
+
ignore_empty_annotations: false
|
164 |
+
transforms:
|
165 |
+
- DetectionPaddedRescale:
|
166 |
+
input_dim: ${dataset_params.val_dataset_params.input_dim}
|
167 |
+
pad_value: 114
|
168 |
+
- DetectionStandardize:
|
169 |
+
max_value: 255.0
|
170 |
+
- DetectionTargetsFormatTransform:
|
171 |
+
input_dim: ${dataset_params.val_dataset_params.input_dim}
|
172 |
+
output_format: LABEL_CXCYWH
|
173 |
+
class_inclusion_list: null
|
174 |
+
max_num_samples: null
|
175 |
+
with_crowd: true
|
176 |
+
verbose: 0
|
177 |
+
val_dataloader_params:
|
178 |
+
batch_size: 16
|
179 |
+
num_workers: 4
|
180 |
+
drop_last: false
|
181 |
+
shuffle: false
|
182 |
+
pin_memory: true
|
183 |
+
collate_fn: CrowdDetectionCollateFN
|
184 |
+
_convert_: all
|
185 |
+
checkpoint_params:
|
186 |
+
load_checkpoint: false
|
187 |
+
load_backbone: false
|
188 |
+
checkpoint_path: null
|
189 |
+
external_checkpoint_path: null
|
190 |
+
source_ckpt_folder_name: null
|
191 |
+
strict_load:
|
192 |
+
_target_: super_gradients.training.sg_trainer.StrictLoad
|
193 |
+
value: no_key_matching
|
194 |
+
pretrained_weights: coco
|
195 |
+
checkpoint_num_classes: null
|
196 |
+
arch_params:
|
197 |
+
in_channels: 3
|
198 |
+
backbone:
|
199 |
+
NStageBackbone:
|
200 |
+
stem:
|
201 |
+
YoloNASStem:
|
202 |
+
out_channels: 48
|
203 |
+
stages:
|
204 |
+
- YoloNASStage:
|
205 |
+
out_channels: 96
|
206 |
+
num_blocks: 2
|
207 |
+
activation_type: relu
|
208 |
+
hidden_channels: 64
|
209 |
+
concat_intermediates: true
|
210 |
+
- YoloNASStage:
|
211 |
+
out_channels: 192
|
212 |
+
num_blocks: 3
|
213 |
+
activation_type: relu
|
214 |
+
hidden_channels: 128
|
215 |
+
concat_intermediates: true
|
216 |
+
- YoloNASStage:
|
217 |
+
out_channels: 384
|
218 |
+
num_blocks: 5
|
219 |
+
activation_type: relu
|
220 |
+
hidden_channels: 256
|
221 |
+
concat_intermediates: true
|
222 |
+
- YoloNASStage:
|
223 |
+
out_channels: 768
|
224 |
+
num_blocks: 2
|
225 |
+
activation_type: relu
|
226 |
+
hidden_channels: 384
|
227 |
+
concat_intermediates: false
|
228 |
+
context_module:
|
229 |
+
SPP:
|
230 |
+
output_channels: 768
|
231 |
+
activation_type: relu
|
232 |
+
k:
|
233 |
+
- 5
|
234 |
+
- 9
|
235 |
+
- 13
|
236 |
+
out_layers:
|
237 |
+
- stage1
|
238 |
+
- stage2
|
239 |
+
- stage3
|
240 |
+
- context_module
|
241 |
+
neck:
|
242 |
+
YoloNASPANNeckWithC2:
|
243 |
+
neck1:
|
244 |
+
YoloNASUpStage:
|
245 |
+
out_channels: 192
|
246 |
+
num_blocks: 2
|
247 |
+
hidden_channels: 192
|
248 |
+
width_mult: 1
|
249 |
+
depth_mult: 1
|
250 |
+
activation_type: relu
|
251 |
+
reduce_channels: true
|
252 |
+
neck2:
|
253 |
+
YoloNASUpStage:
|
254 |
+
out_channels: 96
|
255 |
+
num_blocks: 3
|
256 |
+
hidden_channels: 64
|
257 |
+
width_mult: 1
|
258 |
+
depth_mult: 1
|
259 |
+
activation_type: relu
|
260 |
+
reduce_channels: true
|
261 |
+
neck3:
|
262 |
+
YoloNASDownStage:
|
263 |
+
out_channels: 192
|
264 |
+
num_blocks: 2
|
265 |
+
hidden_channels: 192
|
266 |
+
activation_type: relu
|
267 |
+
width_mult: 1
|
268 |
+
depth_mult: 1
|
269 |
+
neck4:
|
270 |
+
YoloNASDownStage:
|
271 |
+
out_channels: 384
|
272 |
+
num_blocks: 3
|
273 |
+
hidden_channels: 256
|
274 |
+
activation_type: relu
|
275 |
+
width_mult: 1
|
276 |
+
depth_mult: 1
|
277 |
+
heads:
|
278 |
+
NDFLHeads:
|
279 |
+
num_classes: 80
|
280 |
+
reg_max: 16
|
281 |
+
heads_list:
|
282 |
+
- YoloNASDFLHead:
|
283 |
+
inter_channels: 128
|
284 |
+
width_mult: 0.75
|
285 |
+
first_conv_group_size: 0
|
286 |
+
stride: 8
|
287 |
+
- YoloNASDFLHead:
|
288 |
+
inter_channels: 256
|
289 |
+
width_mult: 0.75
|
290 |
+
first_conv_group_size: 0
|
291 |
+
stride: 16
|
292 |
+
- YoloNASDFLHead:
|
293 |
+
inter_channels: 512
|
294 |
+
width_mult: 0.75
|
295 |
+
first_conv_group_size: 0
|
296 |
+
stride: 32
|
297 |
+
bn_eps: 0.001
|
298 |
+
bn_momentum: 0.03
|
299 |
+
inplace_act: true
|
300 |
+
_convert_: all
|
301 |
+
num_classes: ${num_classes}
|
302 |
+
train_dataloader: roboflow_train_yolox
|
303 |
+
val_dataloader: roboflow_val_yolox
|
304 |
+
dataset_name: final-final-c2j0n-mdjfm/3
|
305 |
+
num_classes: 7
|
306 |
+
architecture: yolo_nas_m
|
307 |
+
load_checkpoint: false
|
308 |
+
result_path: null
|
309 |
+
resume: null
|
310 |
+
multi_gpu: false
|
311 |
+
num_gpus: 1
|
312 |
+
experiment_suffix: ''
|
313 |
+
experiment_name: ${architecture}_roboflow_${dataset_name}${experiment_suffix}
|
314 |
+
lr: 0.0004
|
315 |
+
batch_size: 12
|
316 |
+
val_batch_size: 16
|
317 |
+
ema: true
|
318 |
+
epochs: 100
|
319 |
+
num_workers: 4
|
320 |
+
ckpt_root_dir: null
|
.hydra/hydra.yaml
ADDED
@@ -0,0 +1,160 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
hydra:
|
2 |
+
run:
|
3 |
+
dir: ${hydra_output_dir:${ckpt_root_dir}, ${experiment_name}}
|
4 |
+
sweep:
|
5 |
+
dir: multirun/${now:%Y-%m-%d}/${now:%H-%M-%S}
|
6 |
+
subdir: ${hydra.job.num}
|
7 |
+
launcher:
|
8 |
+
_target_: hydra._internal.core_plugins.basic_launcher.BasicLauncher
|
9 |
+
sweeper:
|
10 |
+
_target_: hydra._internal.core_plugins.basic_sweeper.BasicSweeper
|
11 |
+
max_batch_size: null
|
12 |
+
params: null
|
13 |
+
help:
|
14 |
+
app_name: ${hydra.job.name}
|
15 |
+
header: '${hydra.help.app_name} is powered by Hydra.
|
16 |
+
|
17 |
+
'
|
18 |
+
footer: 'Powered by Hydra (https://hydra.cc)
|
19 |
+
|
20 |
+
Use --hydra-help to view Hydra specific help
|
21 |
+
|
22 |
+
'
|
23 |
+
template: '${hydra.help.header}
|
24 |
+
|
25 |
+
== Configuration groups ==
|
26 |
+
|
27 |
+
Compose your configuration from those groups (group=option)
|
28 |
+
|
29 |
+
|
30 |
+
$APP_CONFIG_GROUPS
|
31 |
+
|
32 |
+
|
33 |
+
== Config ==
|
34 |
+
|
35 |
+
Override anything in the config (foo.bar=value)
|
36 |
+
|
37 |
+
|
38 |
+
$CONFIG
|
39 |
+
|
40 |
+
|
41 |
+
${hydra.help.footer}
|
42 |
+
|
43 |
+
'
|
44 |
+
hydra_help:
|
45 |
+
template: 'Hydra (${hydra.runtime.version})
|
46 |
+
|
47 |
+
See https://hydra.cc for more info.
|
48 |
+
|
49 |
+
|
50 |
+
== Flags ==
|
51 |
+
|
52 |
+
$FLAGS_HELP
|
53 |
+
|
54 |
+
|
55 |
+
== Configuration groups ==
|
56 |
+
|
57 |
+
Compose your configuration from those groups (For example, append hydra/job_logging=disabled
|
58 |
+
to command line)
|
59 |
+
|
60 |
+
|
61 |
+
$HYDRA_CONFIG_GROUPS
|
62 |
+
|
63 |
+
|
64 |
+
Use ''--cfg hydra'' to Show the Hydra config.
|
65 |
+
|
66 |
+
'
|
67 |
+
hydra_help: ???
|
68 |
+
hydra_logging:
|
69 |
+
version: 1
|
70 |
+
formatters:
|
71 |
+
simple:
|
72 |
+
format: '[%(asctime)s][HYDRA] %(message)s'
|
73 |
+
handlers:
|
74 |
+
console:
|
75 |
+
class: logging.StreamHandler
|
76 |
+
formatter: simple
|
77 |
+
stream: ext://sys.stdout
|
78 |
+
root:
|
79 |
+
level: INFO
|
80 |
+
handlers:
|
81 |
+
- console
|
82 |
+
loggers:
|
83 |
+
logging_example:
|
84 |
+
level: DEBUG
|
85 |
+
disable_existing_loggers: false
|
86 |
+
job_logging:
|
87 |
+
version: 1
|
88 |
+
formatters:
|
89 |
+
simple:
|
90 |
+
format: '[%(asctime)s][%(name)s][%(levelname)s] - %(message)s'
|
91 |
+
handlers:
|
92 |
+
console:
|
93 |
+
class: logging.StreamHandler
|
94 |
+
formatter: simple
|
95 |
+
stream: ext://sys.stdout
|
96 |
+
file:
|
97 |
+
class: logging.FileHandler
|
98 |
+
formatter: simple
|
99 |
+
filename: ${hydra.runtime.output_dir}/${hydra.job.name}.log
|
100 |
+
root:
|
101 |
+
level: INFO
|
102 |
+
handlers:
|
103 |
+
- console
|
104 |
+
- file
|
105 |
+
disable_existing_loggers: false
|
106 |
+
env: {}
|
107 |
+
mode: RUN
|
108 |
+
searchpath: []
|
109 |
+
callbacks:
|
110 |
+
shortcuts_cb:
|
111 |
+
_target_: super_gradients.common.environment.omegaconf_utils.RecipeShortcutsCallback
|
112 |
+
output_subdir: .hydra
|
113 |
+
overrides:
|
114 |
+
hydra:
|
115 |
+
- hydra.mode=RUN
|
116 |
+
task: []
|
117 |
+
job:
|
118 |
+
name: train_from_recipe
|
119 |
+
chdir: null
|
120 |
+
override_dirname: ''
|
121 |
+
id: ???
|
122 |
+
num: ???
|
123 |
+
config_name: roboflow_yolo_nas_m
|
124 |
+
env_set: {}
|
125 |
+
env_copy: []
|
126 |
+
config:
|
127 |
+
override_dirname:
|
128 |
+
kv_sep: '='
|
129 |
+
item_sep: ','
|
130 |
+
exclude_keys: []
|
131 |
+
runtime:
|
132 |
+
version: 1.3.2
|
133 |
+
version_base: '1.2'
|
134 |
+
cwd: /opt/conda/envs/app/lib/python3.10/site-packages/super_gradients/recipes
|
135 |
+
config_sources:
|
136 |
+
- path: hydra.conf
|
137 |
+
schema: pkg
|
138 |
+
provider: hydra
|
139 |
+
- path: /opt/conda/envs/app/lib/python3.10/site-packages/super_gradients/recipes
|
140 |
+
schema: file
|
141 |
+
provider: main
|
142 |
+
- path: ''
|
143 |
+
schema: structured
|
144 |
+
provider: schema
|
145 |
+
output_dir: /opt/conda/envs/app/lib/python3.10/checkpoints/yolo_nas_m_roboflow_final-final-c2j0n-mdjfm/3
|
146 |
+
choices:
|
147 |
+
arch_params: yolo_nas_m_arch_params
|
148 |
+
checkpoint_params: default_checkpoint_params
|
149 |
+
dataset_params: roboflow_detection_dataset_params
|
150 |
+
training_hyperparams: coco2017_yolo_nas_train_params
|
151 |
+
hydra/env: default
|
152 |
+
hydra/callbacks: null
|
153 |
+
hydra/job_logging: default
|
154 |
+
hydra/hydra_logging: default
|
155 |
+
hydra/hydra_help: default
|
156 |
+
hydra/help: default
|
157 |
+
hydra/sweeper: basic
|
158 |
+
hydra/launcher: basic
|
159 |
+
hydra/output: default
|
160 |
+
verbose: false
|
.hydra/overrides.yaml
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
[]
|
average_model.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:33bcf5ed7d2f930a4ccc3971bd1693f4e5a4333fe8b56df0b674bbd56bde513c
|
3 |
+
size 205085470
|
ckpt_best.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:91ce4e01af7d79d8ca938c5bdf526bbd3fa1a7425155eed177f9492f0ef86008
|
3 |
+
size 681092674
|
ckpt_epoch_100.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6a1089aa072ef28c8db6c0c954a136ce88fe71224ba6b44164194df49b7eb209
|
3 |
+
size 681109574
|
ckpt_latest.pth
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3f76a6064a2733599db3f571ae5d2074d882111d35cbe72882de58e27a0752a7
|
3 |
+
size 681099434
|
console_Nov21_05_01_06.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|
events.out.tfevents.1732165266.1213f0dc8cb8.11180.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:263ea174f666df5d2cde6f0d610ae36fc8d1f4ef08dce4310fde871fda1d5e28
|
3 |
+
size 302121
|
experiment_logs_Nov21_05_01_06.txt
ADDED
The diff for this file is too large to render.
See raw diff
|
|
logs_Nov21_05_01_06.txt
ADDED
@@ -0,0 +1,46 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
[2024-11-21 05:01:02] INFO - super_gradients.common.crash_handler.crash_tips_setup - Crash tips is enabled. You can set your environment variable to CRASH_HANDLER=FALSE to disable it
|
2 |
+
[2024-11-21 05:01:03] DEBUG - matplotlib - matplotlib data path: /opt/conda/envs/app/lib/python3.10/site-packages/matplotlib/mpl-data
|
3 |
+
[2024-11-21 05:01:03] DEBUG - matplotlib - CONFIGDIR=/root/.config/matplotlib
|
4 |
+
[2024-11-21 05:01:03] DEBUG - matplotlib - interactive is False
|
5 |
+
[2024-11-21 05:01:03] DEBUG - matplotlib - platform is linux
|
6 |
+
[2024-11-21 05:01:03] DEBUG - matplotlib - CACHEDIR=/root/.cache/matplotlib
|
7 |
+
[2024-11-21 05:01:03] DEBUG - matplotlib.font_manager - Using fontManager instance from /root/.cache/matplotlib/fontlist-v390.json
|
8 |
+
[2024-11-21 05:01:03] DEBUG - super_gradients.common.sg_loggers.clearml_sg_logger - Failed to import clearml
|
9 |
+
[2024-11-21 05:01:04] DEBUG - hydra.core.utils - Setting JobRuntime:name=UNKNOWN_NAME
|
10 |
+
[2024-11-21 05:01:04] DEBUG - hydra.core.utils - Setting JobRuntime:name=app
|
11 |
+
[2024-11-21 05:01:04] DEBUG - hydra.core.utils - Setting JobRuntime:name=app
|
12 |
+
[2024-11-21 05:01:04] INFO - super_gradients.sanity_check.env_sanity_check - Library check is not supported when super_gradients installed through "git+https://github.com/..." command
|
13 |
+
[2024-11-21 05:01:04] DEBUG - hydra.core.utils - Setting JobRuntime:name=train_from_recipe
|
14 |
+
[2024-11-21 05:01:06] INFO - super_gradients.training.sg_trainer.sg_trainer - Using EMA with params {'decay': 0.9, 'decay_type': 'threshold', 'beta': 15}
|
15 |
+
[2024-11-21 05:01:08] INFO - super_gradients.training.utils.sg_trainer_utils - TRAINING PARAMETERS:
|
16 |
+
- Mode: OFF
|
17 |
+
- Number of GPUs: 1 (1 available on the machine)
|
18 |
+
- Full dataset size: 2399 (len(train_set))
|
19 |
+
- Batch size per GPU: 12 (batch_size)
|
20 |
+
- Batch Accumulate: 1 (batch_accumulate)
|
21 |
+
- Total batch size: 12 (num_gpus * batch_size)
|
22 |
+
- Effective Batch size: 12 (num_gpus * batch_size * batch_accumulate)
|
23 |
+
- Iterations per epoch: 200 (len(train_loader))
|
24 |
+
- Gradient updates per epoch: 200 (len(train_loader) / batch_accumulate)
|
25 |
+
- Model: YoloNAS_M (51.13M parameters, 51.13M optimized)
|
26 |
+
- Learning Rates and Weight Decays:
|
27 |
+
- default: (51.13M parameters). LR: 0.0004 (51.13M parameters) WD: 0.0, (72.22K parameters), WD: 0.0001, (51.06M parameters)
|
28 |
+
|
29 |
+
[2024-11-21 05:01:08] INFO - super_gradients.training.sg_trainer.sg_trainer - Started training for 100 epochs (0/99)
|
30 |
+
|
31 |
+
[2024-11-21 05:01:47] INFO - super_gradients.common.sg_loggers.base_sg_logger - Checkpoint saved in /opt/conda/envs/app/lib/python3.10/checkpoints/yolo_nas_m_roboflow_final-final-c2j0n-mdjfm/3/RUN_20241121_050106_668266/ckpt_best.pth
|
32 |
+
[2024-11-21 05:01:47] INFO - super_gradients.training.sg_trainer.sg_trainer - Best checkpoint overriden: validation mAP@0.50: 0.0021471609361469746
|
33 |
+
[2024-11-21 05:02:25] INFO - super_gradients.common.sg_loggers.base_sg_logger - Checkpoint saved in /opt/conda/envs/app/lib/python3.10/checkpoints/yolo_nas_m_roboflow_final-final-c2j0n-mdjfm/3/RUN_20241121_050106_668266/ckpt_best.pth
|
34 |
+
[2024-11-21 05:02:25] INFO - super_gradients.training.sg_trainer.sg_trainer - Best checkpoint overriden: validation mAP@0.50: 0.820346474647522
|
35 |
+
[2024-11-21 05:03:04] INFO - super_gradients.common.sg_loggers.base_sg_logger - Checkpoint saved in /opt/conda/envs/app/lib/python3.10/checkpoints/yolo_nas_m_roboflow_final-final-c2j0n-mdjfm/3/RUN_20241121_050106_668266/ckpt_best.pth
|
36 |
+
[2024-11-21 05:03:04] INFO - super_gradients.training.sg_trainer.sg_trainer - Best checkpoint overriden: validation mAP@0.50: 0.8461349606513977
|
37 |
+
[2024-11-21 05:05:43] INFO - super_gradients.common.sg_loggers.base_sg_logger - Checkpoint saved in /opt/conda/envs/app/lib/python3.10/checkpoints/yolo_nas_m_roboflow_final-final-c2j0n-mdjfm/3/RUN_20241121_050106_668266/ckpt_best.pth
|
38 |
+
[2024-11-21 05:05:43] INFO - super_gradients.training.sg_trainer.sg_trainer - Best checkpoint overriden: validation mAP@0.50: 0.8567641973495483
|
39 |
+
[2024-11-21 05:08:28] INFO - super_gradients.common.sg_loggers.base_sg_logger - Checkpoint saved in /opt/conda/envs/app/lib/python3.10/checkpoints/yolo_nas_m_roboflow_final-final-c2j0n-mdjfm/3/RUN_20241121_050106_668266/ckpt_best.pth
|
40 |
+
[2024-11-21 05:08:28] INFO - super_gradients.training.sg_trainer.sg_trainer - Best checkpoint overriden: validation mAP@0.50: 0.8589617013931274
|
41 |
+
[2024-11-21 05:15:26] INFO - super_gradients.common.sg_loggers.base_sg_logger - Checkpoint saved in /opt/conda/envs/app/lib/python3.10/checkpoints/yolo_nas_m_roboflow_final-final-c2j0n-mdjfm/3/RUN_20241121_050106_668266/ckpt_best.pth
|
42 |
+
[2024-11-21 05:15:26] INFO - super_gradients.training.sg_trainer.sg_trainer - Best checkpoint overriden: validation mAP@0.50: 0.8630316257476807
|
43 |
+
[2024-11-21 05:39:03] INFO - super_gradients.common.sg_loggers.base_sg_logger - Checkpoint saved in /opt/conda/envs/app/lib/python3.10/checkpoints/yolo_nas_m_roboflow_final-final-c2j0n-mdjfm/3/RUN_20241121_050106_668266/ckpt_best.pth
|
44 |
+
[2024-11-21 05:39:03] INFO - super_gradients.training.sg_trainer.sg_trainer - Best checkpoint overriden: validation mAP@0.50: 0.8690835237503052
|
45 |
+
[2024-11-21 06:10:22] INFO - super_gradients.training.sg_trainer.sg_trainer - RUNNING ADDITIONAL TEST ON THE AVERAGED MODEL...
|
46 |
+
[2024-11-21 06:10:27] INFO - super_gradients.common.sg_loggers.base_sg_logger - [CLEANUP] - Successfully stopped system monitoring process
|