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
|