Initial model and relevant metadata upload

best.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c8b95c35b8981c7301ff04b673b6edbc76af2968dc170ba8bda9341f809eb0b1
+size 1745940171

training_config.yaml

@@ -0,0 +1,194 @@
+data_config:
+  train_labels_path:
+  - combined_forewings.slp
+  val_labels_path: null
+  validation_fraction: 0.1
+  test_file_path: null
+  provider: LabelsReader
+  user_instances_only: true
+  data_pipeline_fw: torch_dataset
+  cache_img_path: null
+  use_existing_imgs: false
+  delete_cache_imgs_after_training: true
+  preprocessing:
+    ensure_rgb: true
+    ensure_grayscale: false
+    max_height: 512
+    max_width: 512
+    scale: 1.0
+    crop_size: null
+    min_crop_size: 100
+  use_augmentations_train: true
+  augmentation_config:
+    intensity:
+      uniform_noise_min: 0.0
+      uniform_noise_max: 1.0
+      uniform_noise_p: 0.0
+      gaussian_noise_mean: 0.0
+      gaussian_noise_std: 1.0
+      gaussian_noise_p: 0.0
+      contrast_min: 0.5
+      contrast_max: 2.0
+      contrast_p: 0.2
+      brightness_min: 0.0
+      brightness_max: 2.0
+      brightness_p: 0.2
+    geometric:
+      rotation_min: -15.0
+      rotation_max: 15.0
+      scale_min: 0.9
+      scale_max: 1.1
+      translate_width: 0.0
+      translate_height: 0.0
+      affine_p: 0.5
+      erase_scale_min: 0.0001
+      erase_scale_max: 0.01
+      erase_ratio_min: 1.0
+      erase_ratio_max: 1.0
+      erase_p: 0.0
+      mixup_lambda_min: 0.01
+      mixup_lambda_max: 0.05
+      mixup_p: 0.0
+  skeletons:
+  - nodes:
+    - name: '1'
+    - name: '2'
+    - name: '3'
+    - name: '4'
+    - name: '5'
+    - name: '6'
+    - name: '7'
+    - name: '8'
+    - name: '9'
+    - name: '10'
+    - name: '11'
+    - name: '12'
+    - name: '13'
+    - name: '14'
+    - name: '15'
+    - name: '16'
+    - name: '17'
+    - name: '18'
+    edges: []
+    symmetries: []
+    name: null
+model_config:
+  init_weights: default
+  pretrained_backbone_weights: null
+  pretrained_head_weights: null
+  backbone_config:
+    unet: null
+    convnext:
+      pre_trained_weights: ConvNeXt_Large_Weights
+      model_type: large
+      arch: null
+      stem_patch_kernel: 4
+      stem_patch_stride: 2
+      in_channels: 3
+      kernel_size: 3
+      filters_rate: 2.0
+      convs_per_block: 2
+      up_interpolate: true
+      output_stride: 2
+      max_stride: 32
+    swint: null
+  head_configs:
+    single_instance:
+      confmaps:
+        part_names:
+        - '1'
+        - '2'
+        - '3'
+        - '4'
+        - '5'
+        - '6'
+        - '7'
+        - '8'
+        - '9'
+        - '10'
+        - '11'
+        - '12'
+        - '13'
+        - '14'
+        - '15'
+        - '16'
+        - '17'
+        - '18'
+        sigma: 3.0
+        output_stride: 2
+    centroid: null
+    centered_instance: null
+    bottomup: null
+    multi_class_bottomup: null
+    multi_class_topdown: null
+  total_params: 436448466
+trainer_config:
+  train_data_loader:
+    batch_size: 2
+    num_workers: 0
+    shuffle: true
+  val_data_loader:
+    batch_size: 2
+    num_workers: 0
+    shuffle: false
+  model_ckpt:
+    save_top_k: 1
+    save_last: false
+  trainer_devices: 1
+  trainer_device_indices: null
+  trainer_accelerator: cuda
+  profiler: null
+  trainer_strategy: auto
+  enable_progress_bar: true
+  min_train_steps_per_epoch: 200
+  train_steps_per_epoch: 344
+  visualize_preds_during_training: true
+  keep_viz: true
+  max_epochs: 200
+  seed: 42
+  use_wandb: false
+  save_ckpt: true
+  ckpt_dir: hp_sweep/models/hp_convnext_large_os2_0.00005lr_2bs
+  run_name: hp_convnext_large_os2_0.00005lr_2bs
+  resume_ckpt_path: null
+  wandb:
+    entity: null
+    project: null
+    name: null
+    save_viz_imgs_wandb: false
+    api_key: null
+    wandb_mode: null
+    prv_runid: null
+    group: null
+    current_run_id: null
+  optimizer_name: Adam
+  optimizer:
+    lr: 5.0e-05
+    amsgrad: false
+  lr_scheduler:
+    step_lr: null
+    reduce_lr_on_plateau:
+      threshold: 1.0e-06
+      threshold_mode: abs
+      cooldown: 3
+      patience: 5
+      factor: 0.5
+      min_lr: 1.0e-08
+  early_stopping:
+    min_delta: 1.0e-08
+    patience: 10
+    stop_training_on_plateau: true
+  online_hard_keypoint_mining:
+    online_mining: false
+    hard_to_easy_ratio: 2.0
+    min_hard_keypoints: 2
+    max_hard_keypoints: null
+    loss_scale: 5.0
+  zmq:
+    controller_port: null
+    controller_polling_timeout: 10
+    publish_port: null
+name: ''
+description: ''
+sleap_nn_version: 0.0.5
+filename: ''

training_log.csv

@@ -0,0 +1,19 @@
+epoch,train_loss,val_loss,learning_rate,train_time,val_time,0,1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17
+0,,0.0017597884871065617,4.999999873689376e-05,,,,,,,,,,,,,,,,,,,,
+0,0.0001435687590856105,0.0001443740475224331,5.000000601285137e-05,,,2.4225937522714958e-05,6.273488543229178e-05,6.087789370212704e-05,0.00014560282579623163,0.00014529646432492882,6.517957808682695e-05,0.00010166212450712919,0.00010492846195120364,0.00014128003385849297,0.0001322729222010821,0.0004002334317192435,0.0002430336899124086,0.0003044233890250325,0.00016710942145437002,8.572812657803297e-05,0.00021965398627799004,0.00010141164239030331,7.858294702600688e-05
+1,6.819902046117932e-05,0.00010251648200210184,5.000000601285137e-05,583.0848388671875,15.469268798828125,1.3722406947636046e-05,2.7804533601738513e-05,9.727877113618888e-06,2.8339043637970462e-05,9.219327330356464e-05,0.00019945972599089146,6.763623241567984e-05,0.00010814602137543261,2.6643858291208744e-05,1.8897917470894754e-05,0.00018021003052126616,9.534606942906976e-05,0.00011531991185620427,9.987275552703068e-05,3.852873487630859e-05,4.969996734871529e-05,3.391706559341401e-05,2.211697028542403e-05
+2,0.00012667500413954258,9.561733168084174e-05,5.000000601285137e-05,581.464599609375,15.486136436462402,4.6630750148324296e-05,8.690095273777843e-05,3.402553920750506e-05,5.947689351160079e-05,0.00011097846436314285,4.396847361931577e-05,0.0002682834165170789,1.7803356968215667e-05,0.00013171568571124226,0.0006447487976402044,0.00029256826383061707,0.00024877797113731503,6.24153544777073e-05,5.3115971240913495e-05,6.755040521966293e-05,4.734890899271704e-05,3.613780063460581e-05,2.7702992156264372e-05
+3,9.772137855179608e-05,9.490588126936927e-05,5.000000601285137e-05,582.3219604492188,15.443947792053223,3.6809404264204204e-05,5.9175508795306087e-05,2.8370552172418684e-05,3.289776941528544e-05,0.00013351975940167904,0.00035961350658908486,0.0004009087570011616,0.00016455886361654848,2.8609227229026146e-05,1.2811109627364203e-05,6.19099082541652e-05,4.206769062875537e-06,3.8009607123967726e-06,4.388673551147804e-05,0.00012708712893072516,1.933950261445716e-05,3.826064494205639e-05,0.00020321841293480247
+4,7.446653762599453e-05,9.134830179391429e-05,5.000000601285137e-05,582.0698852539062,15.461013793945312,4.2182797187706456e-05,1.7114822185249068e-05,9.648573904996738e-05,2.8565613320097327e-05,4.3594503949861974e-05,0.00032352644484490156,1.1922217709070537e-05,5.258755845716223e-05,8.986676220956724e-06,7.075167377479374e-05,4.2939067498082295e-05,3.469549847068265e-05,0.00033635241561569273,5.652024628943764e-05,3.608567203627899e-05,4.0561513742432e-05,8.51345103001222e-05,1.2390662959660403e-05
+5,6.725253479089588e-05,9.188313561026007e-05,5.000000601285137e-05,583.1316528320312,15.500587463378906,2.0861178199993446e-05,9.331625733466353e-06,1.857053575804457e-05,1.3734925232711248e-05,6.122602644609287e-05,0.0001566017308505252,2.8231628675712273e-05,1.1350956810929347e-05,9.195818165608216e-06,7.796288628014736e-06,0.00011447884025983512,0.0004140143282711506,3.425576142035425e-05,7.048466795822605e-05,4.763116976391757e-06,0.00012468782369978726,7.459837797796354e-05,3.6362107493914664e-05
+6,5.920873081777245e-05,8.915931539377198e-05,5.000000601285137e-05,582.5484008789062,15.45626449584961,1.2397153113852255e-05,7.314617960219039e-06,1.398823042109143e-05,1.955730840563774e-05,6.067353024263866e-05,0.0001462936052121222,2.1389689209172502e-05,2.2650194296147674e-05,7.599694072268903e-05,1.4109285075392108e-05,0.00016003251948859543,0.00014152485528029501,3.5443757951725274e-05,9.463827882427722e-05,4.671473288908601e-05,7.752292003715411e-05,0.00010248243052046746,1.3027001841692254e-05
+7,9.777556988410652e-05,9.358614624943584e-05,5.000000601285137e-05,582.98193359375,15.486759185791016,3.778281461563893e-05,1.254957282981195e-06,1.3568325812229887e-05,6.26677501713857e-05,0.0005359392962418497,3.067170109716244e-05,1.0690301678550895e-05,0.00011683569027809426,8.21096637082519e-06,1.0166050742554944e-05,0.00023771871929056942,7.978932990226895e-05,0.00010207760351477191,1.218780471390346e-05,4.406858715810813e-05,1.9411512766964734e-05,0.00039823554106988013,3.868315980071202e-05
+8,4.660034392145462e-05,9.274360490962863e-05,5.000000601285137e-05,582.9597778320312,15.503826141357422,4.481217183638364e-05,2.7613507427304285e-06,7.724940587650053e-06,6.099608071963303e-06,2.4623483113828115e-05,3.0006602173671126e-05,1.724819594528526e-05,4.6732024202356115e-05,1.972790232684929e-05,8.448330845567398e-06,0.00020838991622440517,0.00022675946820527315,1.3658558600582182e-05,2.565169779700227e-05,3.019661016878672e-05,6.9058733060956e-05,1.1615039511525538e-05,4.5291501010069624e-05
+9,5.988934572087601e-05,9.581169433658943e-05,5.000000601285137e-05,582.7284545898438,15.477926254272461,2.6899620934273116e-05,2.5585333787603304e-05,4.474765773920808e-06,1.960322515515145e-05,0.0002343564701732248,6.808086618548259e-05,0.00011246056237723678,1.3179693269194104e-05,2.1126321371411905e-05,9.780221262190025e-06,3.213235322618857e-05,6.451881199609488e-05,7.562337123090401e-05,0.00011820300278486684,5.13821687491145e-05,2.6580777557683177e-05,0.00014863625983707607,2.538424450904131e-05
+10,4.0004928450798616e-05,9.222345397574827e-05,5.000000601285137e-05,583.8795166015625,15.506933212280273,1.5233653357427102e-05,1.3928347470937297e-05,3.4101331038982607e-06,9.538925041852053e-06,0.0001419489417457953,1.935972977662459e-05,1.5775414794916287e-05,1.6028461686801165e-05,5.062897616880946e-05,5.268663153401576e-05,4.42330347141251e-05,2.246247277071234e-05,9.867636254057288e-05,1.349771991954185e-05,1.2209187843836844e-05,3.280009696027264e-05,6.0577058320632204e-05,9.709350706543773e-05
+11,4.505048127612099e-05,9.32385737542063e-05,5.000000601285137e-05,583.1398315429688,15.520587921142578,5.127970507601276e-05,9.267830137105193e-06,3.3336702472297475e-05,1.362614148092689e-05,7.481953161914134e-06,1.8904518583440222e-05,1.258834163309075e-05,5.268955646897666e-06,3.8613770811934955e-06,3.09828610625118e-05,6.98574076523073e-05,0.00017372435831930488,9.696055349195376e-05,6.69116125209257e-05,0.0001095728439395316,4.2157880670856684e-05,7.807630936440546e-06,5.731798955821432e-05
+12,4.1792478441493586e-05,9.11479364731349e-05,5.000000601285137e-05,583.3719482421875,15.478056907653809,7.866421219659969e-05,6.002543159411289e-06,1.8564334823167883e-05,6.629059043916641e-06,6.829008634667844e-05,1.7048234440153465e-05,3.0101833544904366e-05,3.989709875895642e-05,0.00012737407814711332,9.536529432807583e-06,3.553869464667514e-05,9.004769526654854e-05,3.1448928439203883e-06,1.0861184819077607e-05,1.7062830011127517e-05,4.6219403884606436e-05,5.371099541662261e-05,9.357087401440367e-05
+13,4.16866059822496e-05,9.343338024336845e-05,2.5000003006425686e-05,583.90283203125,15.493841171264648,1.3735941138293128e-05,2.9141683626221493e-06,1.4497852589556715e-06,1.205257467518095e-05,1.640964728721883e-05,1.3225554539531004e-05,4.716229886980727e-05,2.0877172573818825e-05,2.0705034330603667e-05,9.418171430297662e-06,0.0001445365051040426,0.00016587579739280045,0.00012295861961320043,6.7004912125412375e-06,5.096263703308068e-05,1.1990567145403475e-05,8.261673792731017e-05,6.767204467905685e-06
+14,0.00042066286550834775,9.599482291378081e-05,2.5000003006425686e-05,583.105224609375,15.5120849609375,0.0004444498335942626,0.0004280824796296656,0.00043266385910101235,0.000432257482316345,0.00043169662239961326,0.00043311569606885314,0.00043797839316539466,0.00043173928861506283,0.0004363855696283281,0.00043005376937799156,0.0004259105189703405,0.0003833481459878385,0.00029241759330034256,0.0004227015597280115,0.0004208990139886737,0.0004326542548369616,0.00042750174179673195,0.00042807593126781285
+15,3.2408544939244166e-05,9.599053009878844e-05,2.5000003006425686e-05,583.346435546875,15.518906593322754,1.0934011697827373e-05,9.724351457407465e-07,5.330731255526189e-06,2.22774178837426e-05,2.3400063582812436e-05,4.9162263167090714e-05,9.174946171697229e-06,5.318813691701507e-06,2.556236540840473e-05,6.696312993881293e-06,5.996802428853698e-05,6.74809271004051e-05,7.230356277432293e-05,4.3437535168777686e-06,3.4087079257005826e-05,3.2167008612304926e-05,0.00014143313455861062,1.2740939382638317e-05
+16,7.324312264245236e-06,9.589452383806929e-05,2.5000003006425686e-05,583.2179565429688,15.480620384216309,5.563185368373524e-06,7.416785479108512e-07,9.716576414575684e-07,4.7253483899112325e-06,2.5753881800483214e-06,4.643809461413184e-06,3.6618880585592706e-06,7.0818132371641695e-06,2.605374902486801e-05,9.669255632616114e-06,6.628546088904841e-06,6.527200639538933e-06,1.875084126368165e-05,3.972084869019454e-06,5.104817319079302e-06,6.276048225117847e-06,1.3956857401353773e-05,4.933452146360651e-06

val_0_pred_metrics.npz

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:71a92c7372f33661f96bbed8650391765c28736a1a304b07411160b3f7f83d30
+size 15521