Uploading /ephemeral/hossein/output/sam2

checkpoints/checkpoint.pt

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

checkpoints/checkpoint_1.pt

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

checkpoints/checkpoint_10.pt

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

checkpoints/checkpoint_11.pt

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

checkpoints/checkpoint_12.pt

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

checkpoints/checkpoint_13.pt

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

checkpoints/checkpoint_14.pt

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

checkpoints/checkpoint_15.pt

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

checkpoints/checkpoint_16.pt

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

checkpoints/checkpoint_17.pt

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

checkpoints/checkpoint_18.pt

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

checkpoints/checkpoint_19.pt

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

checkpoints/checkpoint_2.pt

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

checkpoints/checkpoint_20.pt

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

checkpoints/checkpoint_21.pt

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

checkpoints/checkpoint_22.pt

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

checkpoints/checkpoint_23.pt

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

checkpoints/checkpoint_24.pt

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

checkpoints/checkpoint_25.pt

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

checkpoints/checkpoint_26.pt

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

checkpoints/checkpoint_27.pt

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

checkpoints/checkpoint_28.pt

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

checkpoints/checkpoint_29.pt

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

checkpoints/checkpoint_3.pt

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

checkpoints/checkpoint_30.pt

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

checkpoints/checkpoint_31.pt

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

checkpoints/checkpoint_32.pt

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

checkpoints/checkpoint_33.pt

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

checkpoints/checkpoint_34.pt

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

checkpoints/checkpoint_35.pt

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

checkpoints/checkpoint_36.pt

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

checkpoints/checkpoint_37.pt

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

checkpoints/checkpoint_38.pt

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

checkpoints/checkpoint_39.pt

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

checkpoints/checkpoint_4.pt

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

checkpoints/checkpoint_40.pt

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

checkpoints/checkpoint_5.pt

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

checkpoints/checkpoint_6.pt

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

checkpoints/checkpoint_7.pt

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

checkpoints/checkpoint_8.pt

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

checkpoints/checkpoint_9.pt

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

config.yaml

@@ -0,0 +1,330 @@
+scratch:
+  resolution: 1024
+  train_batch_size: 1
+  num_train_workers: 3
+  num_frames: 8
+  max_num_objects: 4
+  base_lr: 5.0e-06
+  vision_lr: 3.0e-06
+  phases_per_epoch: 1
+  num_epochs: 40
+dataset:
+  img_folder: /home/hossein/hossein/projects/sam2/training/ptmc-data/JPEGImages
+  gt_folder: /home/hossein/hossein/projects/sam2/training/ptmc-data/Annotations
+  file_list_txt: null
+  multiplier: 2
+vos:
+  train_transforms:
+  - _target_: training.dataset.transforms.ComposeAPI
+    transforms:
+    - _target_: training.dataset.transforms.RandomHorizontalFlip
+      consistent_transform: true
+    - _target_: training.dataset.transforms.RandomAffine
+      degrees: 25
+      shear: 20
+      image_interpolation: bilinear
+      consistent_transform: true
+    - _target_: training.dataset.transforms.RandomResizeAPI
+      sizes: ${scratch.resolution}
+      square: true
+      consistent_transform: true
+    - _target_: training.dataset.transforms.ColorJitter
+      consistent_transform: true
+      brightness: 0.1
+      contrast: 0.03
+      saturation: 0.03
+      hue: null
+    - _target_: training.dataset.transforms.RandomGrayscale
+      p: 0.05
+      consistent_transform: true
+    - _target_: training.dataset.transforms.ColorJitter
+      consistent_transform: false
+      brightness: 0.1
+      contrast: 0.05
+      saturation: 0.05
+      hue: null
+    - _target_: training.dataset.transforms.ToTensorAPI
+    - _target_: training.dataset.transforms.NormalizeAPI
+      mean:
+      - 0.485
+      - 0.456
+      - 0.406
+      std:
+      - 0.229
+      - 0.224
+      - 0.225
+trainer:
+  _target_: training.trainer.Trainer
+  mode: train_only
+  max_epochs: ${times:${scratch.num_epochs},${scratch.phases_per_epoch}}
+  accelerator: cuda
+  seed_value: 123
+  model:
+    _target_: training.model.sam2.SAM2Train
+    image_encoder:
+      _target_: sam2.modeling.backbones.image_encoder.ImageEncoder
+      scalp: 1
+      trunk:
+        _target_: sam2.modeling.backbones.hieradet.Hiera
+        embed_dim: 144
+        num_heads: 2
+        stages:
+        - 2
+        - 6
+        - 36
+        - 4
+        global_att_blocks:
+        - 23
+        - 33
+        - 43
+        window_pos_embed_bkg_spatial_size:
+        - 7
+        - 7
+        window_spec:
+        - 8
+        - 4
+        - 16
+        - 8
+      neck:
+        _target_: sam2.modeling.backbones.image_encoder.FpnNeck
+        position_encoding:
+          _target_: sam2.modeling.position_encoding.PositionEmbeddingSine
+          num_pos_feats: 256
+          normalize: true
+          scale: null
+          temperature: 10000
+        d_model: 256
+        backbone_channel_list:
+        - 1152
+        - 576
+        - 288
+        - 144
+        fpn_top_down_levels:
+        - 2
+        - 3
+        fpn_interp_model: nearest
+    memory_attention:
+      _target_: sam2.modeling.memory_attention.MemoryAttention
+      d_model: 256
+      pos_enc_at_input: true
+      layer:
+        _target_: sam2.modeling.memory_attention.MemoryAttentionLayer
+        activation: relu
+        dim_feedforward: 2048
+        dropout: 0.1
+        pos_enc_at_attn: false
+        self_attention:
+          _target_: sam2.modeling.sam.transformer.RoPEAttention
+          rope_theta: 10000.0
+          feat_sizes:
+          - 64
+          - 64
+          embedding_dim: 256
+          num_heads: 1
+          downsample_rate: 1
+          dropout: 0.1
+        d_model: 256
+        pos_enc_at_cross_attn_keys: true
+        pos_enc_at_cross_attn_queries: false
+        cross_attention:
+          _target_: sam2.modeling.sam.transformer.RoPEAttention
+          rope_theta: 10000.0
+          feat_sizes:
+          - 64
+          - 64
+          rope_k_repeat: true
+          embedding_dim: 256
+          num_heads: 1
+          downsample_rate: 1
+          dropout: 0.1
+          kv_in_dim: 64
+      num_layers: 4
+    memory_encoder:
+      _target_: sam2.modeling.memory_encoder.MemoryEncoder
+      out_dim: 64
+      position_encoding:
+        _target_: sam2.modeling.position_encoding.PositionEmbeddingSine
+        num_pos_feats: 64
+        normalize: true
+        scale: null
+        temperature: 10000
+      mask_downsampler:
+        _target_: sam2.modeling.memory_encoder.MaskDownSampler
+        kernel_size: 3
+        stride: 2
+        padding: 1
+      fuser:
+        _target_: sam2.modeling.memory_encoder.Fuser
+        layer:
+          _target_: sam2.modeling.memory_encoder.CXBlock
+          dim: 256
+          kernel_size: 7
+          padding: 3
+          layer_scale_init_value: 1.0e-06
+          use_dwconv: true
+        num_layers: 2
+    num_maskmem: 7
+    image_size: ${scratch.resolution}
+    sigmoid_scale_for_mem_enc: 20.0
+    sigmoid_bias_for_mem_enc: -10.0
+    use_mask_input_as_output_without_sam: true
+    directly_add_no_mem_embed: true
+    no_obj_embed_spatial: true
+    use_high_res_features_in_sam: true
+    multimask_output_in_sam: true
+    iou_prediction_use_sigmoid: true
+    use_obj_ptrs_in_encoder: true
+    add_tpos_enc_to_obj_ptrs: true
+    proj_tpos_enc_in_obj_ptrs: true
+    use_signed_tpos_enc_to_obj_ptrs: true
+    only_obj_ptrs_in_the_past_for_eval: true
+    pred_obj_scores: true
+    pred_obj_scores_mlp: true
+    fixed_no_obj_ptr: true
+    multimask_output_for_tracking: true
+    use_multimask_token_for_obj_ptr: true
+    multimask_min_pt_num: 0
+    multimask_max_pt_num: 1
+    use_mlp_for_obj_ptr_proj: true
+    compile_image_encoder: false
+    prob_to_use_pt_input_for_train: 0.5
+    prob_to_use_pt_input_for_eval: 0.0
+    prob_to_use_box_input_for_train: 0.5
+    prob_to_use_box_input_for_eval: 0.0
+    prob_to_sample_from_gt_for_train: 0.1
+    num_frames_to_correct_for_train: 2
+    num_frames_to_correct_for_eval: 1
+    rand_frames_to_correct_for_train: true
+    add_all_frames_to_correct_as_cond: true
+    num_init_cond_frames_for_train: 2
+    rand_init_cond_frames_for_train: true
+    num_correction_pt_per_frame: 7
+    use_act_ckpt_iterative_pt_sampling: false
+    num_init_cond_frames_for_eval: 1
+    forward_backbone_per_frame_for_eval: true
+  data:
+    train:
+      _target_: training.dataset.sam2_datasets.TorchTrainMixedDataset
+      phases_per_epoch: ${scratch.phases_per_epoch}
+      batch_sizes:
+      - ${scratch.train_batch_size}
+      datasets:
+      - _target_: training.dataset.utils.RepeatFactorWrapper
+        dataset:
+          _target_: training.dataset.utils.ConcatDataset
+          datasets:
+          - _target_: training.dataset.vos_dataset.VOSDataset
+            transforms: ${vos.train_transforms}
+            training: true
+            video_dataset:
+              _target_: training.dataset.vos_raw_dataset.PNGRawDataset
+              img_folder: ${dataset.img_folder}
+              gt_folder: ${dataset.gt_folder}
+              file_list_txt: ${dataset.file_list_txt}
+            sampler:
+              _target_: training.dataset.vos_sampler.RandomUniformSampler
+              num_frames: ${scratch.num_frames}
+              max_num_objects: ${scratch.max_num_objects}
+            multiplier: ${dataset.multiplier}
+      shuffle: true
+      num_workers: ${scratch.num_train_workers}
+      pin_memory: true
+      drop_last: true
+      collate_fn:
+        _target_: training.utils.data_utils.collate_fn
+        _partial_: true
+        dict_key: all
+  optim:
+    amp:
+      enabled: true
+      amp_dtype: bfloat16
+    optimizer:
+      _target_: torch.optim.AdamW
+    gradient_clip:
+      _target_: training.optimizer.GradientClipper
+      max_norm: 0.1
+      norm_type: 2
+    param_group_modifiers:
+    - _target_: training.optimizer.layer_decay_param_modifier
+      _partial_: true
+      layer_decay_value: 0.9
+      apply_to: image_encoder.trunk
+      overrides:
+      - pattern: '*pos_embed*'
+        value: 1.0
+    options:
+      lr:
+      - scheduler:
+          _target_: fvcore.common.param_scheduler.CosineParamScheduler
+          start_value: ${scratch.base_lr}
+          end_value: ${divide:${scratch.base_lr},10}
+      - scheduler:
+          _target_: fvcore.common.param_scheduler.CosineParamScheduler
+          start_value: ${scratch.vision_lr}
+          end_value: ${divide:${scratch.vision_lr},10}
+        param_names:
+        - image_encoder.*
+      weight_decay:
+      - scheduler:
+          _target_: fvcore.common.param_scheduler.ConstantParamScheduler
+          value: 0.1
+      - scheduler:
+          _target_: fvcore.common.param_scheduler.ConstantParamScheduler
+          value: 0.0
+        param_names:
+        - '*bias*'
+        module_cls_names:
+        - torch.nn.LayerNorm
+  loss:
+    all:
+      _target_: training.loss_fns.MultiStepMultiMasksAndIous
+      weight_dict:
+        loss_mask: 20
+        loss_dice: 1
+        loss_iou: 1
+        loss_class: 1
+      supervise_all_iou: true
+      iou_use_l1_loss: true
+      pred_obj_scores: true
+      focal_gamma_obj_score: 0.0
+      focal_alpha_obj_score: -1.0
+  distributed:
+    backend: nccl
+    find_unused_parameters: true
+  logging:
+    tensorboard_writer:
+      _target_: training.utils.logger.make_tensorboard_logger
+      log_dir: ${launcher.experiment_log_dir}/tensorboard
+      flush_secs: 120
+      should_log: true
+    log_dir: ${launcher.experiment_log_dir}/logs
+    log_freq: 10
+  checkpoint:
+    save_dir: ${launcher.experiment_log_dir}/checkpoints
+    save_freq: 1
+    model_weight_initializer:
+      _partial_: true
+      _target_: training.utils.checkpoint_utils.load_state_dict_into_model
+      strict: true
+      ignore_unexpected_keys: null
+      ignore_missing_keys: null
+      state_dict:
+        _target_: training.utils.checkpoint_utils.load_checkpoint_and_apply_kernels
+        checkpoint_path: /home/hossein/hossein/projects/sam2/checkpoints/sam2.1_hiera_large.pt
+        ckpt_state_dict_keys:
+        - model
+launcher:
+  num_nodes: 1
+  gpus_per_node: 4
+  experiment_log_dir: /ephemeral/hossein/output/sam2
+submitit:
+  partition: null
+  account: null
+  qos: null
+  cpus_per_task: 10
+  use_cluster: false
+  timeout_hour: 24
+  name: null
+  port_range:
+  - 10000
+  - 65000

config_resolved.yaml

@@ -0,0 +1,368 @@
+scratch:
+  resolution: 1024
+  train_batch_size: 1
+  num_train_workers: 3
+  num_frames: 8
+  max_num_objects: 4
+  base_lr: 5.0e-06
+  vision_lr: 3.0e-06
+  phases_per_epoch: 1
+  num_epochs: 40
+dataset:
+  img_folder: /home/hossein/hossein/projects/sam2/training/ptmc-data/JPEGImages
+  gt_folder: /home/hossein/hossein/projects/sam2/training/ptmc-data/Annotations
+  file_list_txt: null
+  multiplier: 2
+vos:
+  train_transforms:
+  - _target_: training.dataset.transforms.ComposeAPI
+    transforms:
+    - _target_: training.dataset.transforms.RandomHorizontalFlip
+      consistent_transform: true
+    - _target_: training.dataset.transforms.RandomAffine
+      degrees: 25
+      shear: 20
+      image_interpolation: bilinear
+      consistent_transform: true
+    - _target_: training.dataset.transforms.RandomResizeAPI
+      sizes: 1024
+      square: true
+      consistent_transform: true
+    - _target_: training.dataset.transforms.ColorJitter
+      consistent_transform: true
+      brightness: 0.1
+      contrast: 0.03
+      saturation: 0.03
+      hue: null
+    - _target_: training.dataset.transforms.RandomGrayscale
+      p: 0.05
+      consistent_transform: true
+    - _target_: training.dataset.transforms.ColorJitter
+      consistent_transform: false
+      brightness: 0.1
+      contrast: 0.05
+      saturation: 0.05
+      hue: null
+    - _target_: training.dataset.transforms.ToTensorAPI
+    - _target_: training.dataset.transforms.NormalizeAPI
+      mean:
+      - 0.485
+      - 0.456
+      - 0.406
+      std:
+      - 0.229
+      - 0.224
+      - 0.225
+trainer:
+  _target_: training.trainer.Trainer
+  mode: train_only
+  max_epochs: 40
+  accelerator: cuda
+  seed_value: 123
+  model:
+    _target_: training.model.sam2.SAM2Train
+    image_encoder:
+      _target_: sam2.modeling.backbones.image_encoder.ImageEncoder
+      scalp: 1
+      trunk:
+        _target_: sam2.modeling.backbones.hieradet.Hiera
+        embed_dim: 144
+        num_heads: 2
+        stages:
+        - 2
+        - 6
+        - 36
+        - 4
+        global_att_blocks:
+        - 23
+        - 33
+        - 43
+        window_pos_embed_bkg_spatial_size:
+        - 7
+        - 7
+        window_spec:
+        - 8
+        - 4
+        - 16
+        - 8
+      neck:
+        _target_: sam2.modeling.backbones.image_encoder.FpnNeck
+        position_encoding:
+          _target_: sam2.modeling.position_encoding.PositionEmbeddingSine
+          num_pos_feats: 256
+          normalize: true
+          scale: null
+          temperature: 10000
+        d_model: 256
+        backbone_channel_list:
+        - 1152
+        - 576
+        - 288
+        - 144
+        fpn_top_down_levels:
+        - 2
+        - 3
+        fpn_interp_model: nearest
+    memory_attention:
+      _target_: sam2.modeling.memory_attention.MemoryAttention
+      d_model: 256
+      pos_enc_at_input: true
+      layer:
+        _target_: sam2.modeling.memory_attention.MemoryAttentionLayer
+        activation: relu
+        dim_feedforward: 2048
+        dropout: 0.1
+        pos_enc_at_attn: false
+        self_attention:
+          _target_: sam2.modeling.sam.transformer.RoPEAttention
+          rope_theta: 10000.0
+          feat_sizes:
+          - 64
+          - 64
+          embedding_dim: 256
+          num_heads: 1
+          downsample_rate: 1
+          dropout: 0.1
+        d_model: 256
+        pos_enc_at_cross_attn_keys: true
+        pos_enc_at_cross_attn_queries: false
+        cross_attention:
+          _target_: sam2.modeling.sam.transformer.RoPEAttention
+          rope_theta: 10000.0
+          feat_sizes:
+          - 64
+          - 64
+          rope_k_repeat: true
+          embedding_dim: 256
+          num_heads: 1
+          downsample_rate: 1
+          dropout: 0.1
+          kv_in_dim: 64
+      num_layers: 4
+    memory_encoder:
+      _target_: sam2.modeling.memory_encoder.MemoryEncoder
+      out_dim: 64
+      position_encoding:
+        _target_: sam2.modeling.position_encoding.PositionEmbeddingSine
+        num_pos_feats: 64
+        normalize: true
+        scale: null
+        temperature: 10000
+      mask_downsampler:
+        _target_: sam2.modeling.memory_encoder.MaskDownSampler
+        kernel_size: 3
+        stride: 2
+        padding: 1
+      fuser:
+        _target_: sam2.modeling.memory_encoder.Fuser
+        layer:
+          _target_: sam2.modeling.memory_encoder.CXBlock
+          dim: 256
+          kernel_size: 7
+          padding: 3
+          layer_scale_init_value: 1.0e-06
+          use_dwconv: true
+        num_layers: 2
+    num_maskmem: 7
+    image_size: 1024
+    sigmoid_scale_for_mem_enc: 20.0
+    sigmoid_bias_for_mem_enc: -10.0
+    use_mask_input_as_output_without_sam: true
+    directly_add_no_mem_embed: true
+    no_obj_embed_spatial: true
+    use_high_res_features_in_sam: true
+    multimask_output_in_sam: true
+    iou_prediction_use_sigmoid: true
+    use_obj_ptrs_in_encoder: true
+    add_tpos_enc_to_obj_ptrs: true
+    proj_tpos_enc_in_obj_ptrs: true
+    use_signed_tpos_enc_to_obj_ptrs: true
+    only_obj_ptrs_in_the_past_for_eval: true
+    pred_obj_scores: true
+    pred_obj_scores_mlp: true
+    fixed_no_obj_ptr: true
+    multimask_output_for_tracking: true
+    use_multimask_token_for_obj_ptr: true
+    multimask_min_pt_num: 0
+    multimask_max_pt_num: 1
+    use_mlp_for_obj_ptr_proj: true
+    compile_image_encoder: false
+    prob_to_use_pt_input_for_train: 0.5
+    prob_to_use_pt_input_for_eval: 0.0
+    prob_to_use_box_input_for_train: 0.5
+    prob_to_use_box_input_for_eval: 0.0
+    prob_to_sample_from_gt_for_train: 0.1
+    num_frames_to_correct_for_train: 2
+    num_frames_to_correct_for_eval: 1
+    rand_frames_to_correct_for_train: true
+    add_all_frames_to_correct_as_cond: true
+    num_init_cond_frames_for_train: 2
+    rand_init_cond_frames_for_train: true
+    num_correction_pt_per_frame: 7
+    use_act_ckpt_iterative_pt_sampling: false
+    num_init_cond_frames_for_eval: 1
+    forward_backbone_per_frame_for_eval: true
+  data:
+    train:
+      _target_: training.dataset.sam2_datasets.TorchTrainMixedDataset
+      phases_per_epoch: 1
+      batch_sizes:
+      - 1
+      datasets:
+      - _target_: training.dataset.utils.RepeatFactorWrapper
+        dataset:
+          _target_: training.dataset.utils.ConcatDataset
+          datasets:
+          - _target_: training.dataset.vos_dataset.VOSDataset
+            transforms:
+            - _target_: training.dataset.transforms.ComposeAPI
+              transforms:
+              - _target_: training.dataset.transforms.RandomHorizontalFlip
+                consistent_transform: true
+              - _target_: training.dataset.transforms.RandomAffine
+                degrees: 25
+                shear: 20
+                image_interpolation: bilinear
+                consistent_transform: true
+              - _target_: training.dataset.transforms.RandomResizeAPI
+                sizes: 1024
+                square: true
+                consistent_transform: true
+              - _target_: training.dataset.transforms.ColorJitter
+                consistent_transform: true
+                brightness: 0.1
+                contrast: 0.03
+                saturation: 0.03
+                hue: null
+              - _target_: training.dataset.transforms.RandomGrayscale
+                p: 0.05
+                consistent_transform: true
+              - _target_: training.dataset.transforms.ColorJitter
+                consistent_transform: false
+                brightness: 0.1
+                contrast: 0.05
+                saturation: 0.05
+                hue: null
+              - _target_: training.dataset.transforms.ToTensorAPI
+              - _target_: training.dataset.transforms.NormalizeAPI
+                mean:
+                - 0.485
+                - 0.456
+                - 0.406
+                std:
+                - 0.229
+                - 0.224
+                - 0.225
+            training: true
+            video_dataset:
+              _target_: training.dataset.vos_raw_dataset.PNGRawDataset
+              img_folder: /home/hossein/hossein/projects/sam2/training/ptmc-data/JPEGImages
+              gt_folder: /home/hossein/hossein/projects/sam2/training/ptmc-data/Annotations
+              file_list_txt: null
+            sampler:
+              _target_: training.dataset.vos_sampler.RandomUniformSampler
+              num_frames: 8
+              max_num_objects: 4
+            multiplier: 2
+      shuffle: true
+      num_workers: 3
+      pin_memory: true
+      drop_last: true
+      collate_fn:
+        _target_: training.utils.data_utils.collate_fn
+        _partial_: true
+        dict_key: all
+  optim:
+    amp:
+      enabled: true
+      amp_dtype: bfloat16
+    optimizer:
+      _target_: torch.optim.AdamW
+    gradient_clip:
+      _target_: training.optimizer.GradientClipper
+      max_norm: 0.1
+      norm_type: 2
+    param_group_modifiers:
+    - _target_: training.optimizer.layer_decay_param_modifier
+      _partial_: true
+      layer_decay_value: 0.9
+      apply_to: image_encoder.trunk
+      overrides:
+      - pattern: '*pos_embed*'
+        value: 1.0
+    options:
+      lr:
+      - scheduler:
+          _target_: fvcore.common.param_scheduler.CosineParamScheduler
+          start_value: 5.0e-06
+          end_value: 5.000000000000001e-07
+      - scheduler:
+          _target_: fvcore.common.param_scheduler.CosineParamScheduler
+          start_value: 3.0e-06
+          end_value: 3.0e-07
+        param_names:
+        - image_encoder.*
+      weight_decay:
+      - scheduler:
+          _target_: fvcore.common.param_scheduler.ConstantParamScheduler
+          value: 0.1
+      - scheduler:
+          _target_: fvcore.common.param_scheduler.ConstantParamScheduler
+          value: 0.0
+        param_names:
+        - '*bias*'
+        module_cls_names:
+        - torch.nn.LayerNorm
+  loss:
+    all:
+      _target_: training.loss_fns.MultiStepMultiMasksAndIous
+      weight_dict:
+        loss_mask: 20
+        loss_dice: 1
+        loss_iou: 1
+        loss_class: 1
+      supervise_all_iou: true
+      iou_use_l1_loss: true
+      pred_obj_scores: true
+      focal_gamma_obj_score: 0.0
+      focal_alpha_obj_score: -1.0
+  distributed:
+    backend: nccl
+    find_unused_parameters: true
+  logging:
+    tensorboard_writer:
+      _target_: training.utils.logger.make_tensorboard_logger
+      log_dir: /ephemeral/hossein/output/sam2/tensorboard
+      flush_secs: 120
+      should_log: true
+    log_dir: /ephemeral/hossein/output/sam2/logs
+    log_freq: 10
+  checkpoint:
+    save_dir: /ephemeral/hossein/output/sam2/checkpoints
+    save_freq: 1
+    model_weight_initializer:
+      _partial_: true
+      _target_: training.utils.checkpoint_utils.load_state_dict_into_model
+      strict: true
+      ignore_unexpected_keys: null
+      ignore_missing_keys: null
+      state_dict:
+        _target_: training.utils.checkpoint_utils.load_checkpoint_and_apply_kernels
+        checkpoint_path: /home/hossein/hossein/projects/sam2/checkpoints/sam2.1_hiera_large.pt
+        ckpt_state_dict_keys:
+        - model
+launcher:
+  num_nodes: 1
+  gpus_per_node: 4
+  experiment_log_dir: /ephemeral/hossein/output/sam2
+submitit:
+  partition: null
+  account: null
+  qos: null
+  cpus_per_task: 10
+  use_cluster: false
+  timeout_hour: 24
+  name: null
+  port_range:
+  - 10000
+  - 65000

logs/best_stats.json

@@ -0,0 +1,40 @@
+{"Trainer/where": 0.024285714285714285, "Trainer/epoch": 0, "Trainer/steps_train": 35}
+{"Trainer/where": 0.04928571428571429, "Trainer/epoch": 1, "Trainer/steps_train": 70}
+{"Trainer/where": 0.07428571428571429, "Trainer/epoch": 2, "Trainer/steps_train": 105}
+{"Trainer/where": 0.09928571428571428, "Trainer/epoch": 3, "Trainer/steps_train": 140}
+{"Trainer/where": 0.12428571428571429, "Trainer/epoch": 4, "Trainer/steps_train": 175}
+{"Trainer/where": 0.1492857142857143, "Trainer/epoch": 5, "Trainer/steps_train": 210}
+{"Trainer/where": 0.1742857142857143, "Trainer/epoch": 6, "Trainer/steps_train": 245}
+{"Trainer/where": 0.1992857142857143, "Trainer/epoch": 7, "Trainer/steps_train": 280}
+{"Trainer/where": 0.22428571428571428, "Trainer/epoch": 8, "Trainer/steps_train": 315}
+{"Trainer/where": 0.24928571428571428, "Trainer/epoch": 9, "Trainer/steps_train": 350}
+{"Trainer/where": 0.2742857142857143, "Trainer/epoch": 10, "Trainer/steps_train": 385}
+{"Trainer/where": 0.29928571428571427, "Trainer/epoch": 11, "Trainer/steps_train": 420}
+{"Trainer/where": 0.3242857142857143, "Trainer/epoch": 12, "Trainer/steps_train": 455}
+{"Trainer/where": 0.3492857142857143, "Trainer/epoch": 13, "Trainer/steps_train": 490}
+{"Trainer/where": 0.3742857142857143, "Trainer/epoch": 14, "Trainer/steps_train": 525}
+{"Trainer/where": 0.3992857142857143, "Trainer/epoch": 15, "Trainer/steps_train": 560}
+{"Trainer/where": 0.42428571428571427, "Trainer/epoch": 16, "Trainer/steps_train": 595}
+{"Trainer/where": 0.4492857142857143, "Trainer/epoch": 17, "Trainer/steps_train": 630}
+{"Trainer/where": 0.4742857142857143, "Trainer/epoch": 18, "Trainer/steps_train": 665}
+{"Trainer/where": 0.4992857142857143, "Trainer/epoch": 19, "Trainer/steps_train": 700}
+{"Trainer/where": 0.5242857142857142, "Trainer/epoch": 20, "Trainer/steps_train": 735}
+{"Trainer/where": 0.5492857142857143, "Trainer/epoch": 21, "Trainer/steps_train": 770}
+{"Trainer/where": 0.5742857142857143, "Trainer/epoch": 22, "Trainer/steps_train": 805}
+{"Trainer/where": 0.5992857142857143, "Trainer/epoch": 23, "Trainer/steps_train": 840}
+{"Trainer/where": 0.6242857142857143, "Trainer/epoch": 24, "Trainer/steps_train": 875}
+{"Trainer/where": 0.6492857142857142, "Trainer/epoch": 25, "Trainer/steps_train": 910}
+{"Trainer/where": 0.6742857142857143, "Trainer/epoch": 26, "Trainer/steps_train": 945}
+{"Trainer/where": 0.6992857142857143, "Trainer/epoch": 27, "Trainer/steps_train": 980}
+{"Trainer/where": 0.7242857142857143, "Trainer/epoch": 28, "Trainer/steps_train": 1015}
+{"Trainer/where": 0.7492857142857143, "Trainer/epoch": 29, "Trainer/steps_train": 1050}
+{"Trainer/where": 0.7742857142857142, "Trainer/epoch": 30, "Trainer/steps_train": 1085}
+{"Trainer/where": 0.7992857142857143, "Trainer/epoch": 31, "Trainer/steps_train": 1120}
+{"Trainer/where": 0.8242857142857142, "Trainer/epoch": 32, "Trainer/steps_train": 1155}
+{"Trainer/where": 0.8492857142857142, "Trainer/epoch": 33, "Trainer/steps_train": 1190}
+{"Trainer/where": 0.8742857142857142, "Trainer/epoch": 34, "Trainer/steps_train": 1225}
+{"Trainer/where": 0.8992857142857142, "Trainer/epoch": 35, "Trainer/steps_train": 1260}
+{"Trainer/where": 0.9242857142857142, "Trainer/epoch": 36, "Trainer/steps_train": 1295}
+{"Trainer/where": 0.9492857142857142, "Trainer/epoch": 37, "Trainer/steps_train": 1330}
+{"Trainer/where": 0.9742857142857142, "Trainer/epoch": 38, "Trainer/steps_train": 1365}
+{"Trainer/where": 0.9992857142857142, "Trainer/epoch": 39, "Trainer/steps_train": 1400}

logs/train_stats.json

@@ -0,0 +1,40 @@
+{"Losses/train_all_loss": 15.52312662942069, "Losses/train_all_loss_mask": 0.03629592404467985, "Losses/train_all_loss_dice": 10.42937673841204, "Losses/train_all_loss_iou": 1.4414258205597954, "Losses/train_all_loss_class": 2.9264054538948194, "Losses/train_all_core_loss": 15.52312662942069, "Trainer/where": 0.024285714285714285, "Trainer/epoch": 0, "Trainer/steps_train": 35}
+{"Losses/train_all_loss": 13.35472868510655, "Losses/train_all_loss_mask": 0.021235573248954358, "Losses/train_all_loss_dice": 10.641365950448172, "Losses/train_all_loss_iou": 0.7040971059019544, "Losses/train_all_loss_class": 1.5845541834831238, "Losses/train_all_core_loss": 13.35472868510655, "Trainer/where": 0.04928571428571429, "Trainer/epoch": 1, "Trainer/steps_train": 70}
+{"Losses/train_all_loss": 13.056742286682129, "Losses/train_all_loss_mask": 0.011416948284854048, "Losses/train_all_loss_dice": 10.823509979248048, "Losses/train_all_loss_iou": 0.5041234560734925, "Losses/train_all_loss_class": 1.500769633267607, "Losses/train_all_core_loss": 13.056742286682129, "Trainer/where": 0.07428571428571429, "Trainer/epoch": 2, "Trainer/steps_train": 105}
+{"Losses/train_all_loss": 13.252275030953543, "Losses/train_all_loss_mask": 0.0038689142230266173, "Losses/train_all_loss_dice": 10.98888555254255, "Losses/train_all_loss_iou": 0.22835017135366797, "Losses/train_all_loss_class": 1.9576609894633292, "Losses/train_all_core_loss": 13.252275030953543, "Trainer/where": 0.09928571428571428, "Trainer/epoch": 3, "Trainer/steps_train": 140}
+{"Losses/train_all_loss": 13.165853296007429, "Losses/train_all_loss_mask": 0.003397888245152509, "Losses/train_all_loss_dice": 10.91664868763515, "Losses/train_all_loss_iou": 0.17255155175392117, "Losses/train_all_loss_class": 2.00869502723217, "Losses/train_all_core_loss": 13.165853296007429, "Trainer/where": 0.12428571428571429, "Trainer/epoch": 4, "Trainer/steps_train": 175}
+{"Losses/train_all_loss": 12.23187631879534, "Losses/train_all_loss_mask": 0.003290402367877375, "Losses/train_all_loss_dice": 10.583557428632464, "Losses/train_all_loss_iou": 0.1472343360522895, "Losses/train_all_loss_class": 1.4352764577205692, "Losses/train_all_core_loss": 12.23187631879534, "Trainer/where": 0.1492857142857143, "Trainer/epoch": 5, "Trainer/steps_train": 210}
+{"Losses/train_all_loss": 12.532559653690884, "Losses/train_all_loss_mask": 0.00453987946135125, "Losses/train_all_loss_dice": 10.932144478389196, "Losses/train_all_loss_iou": 0.11682707172585652, "Losses/train_all_loss_class": 1.3927903246666704, "Losses/train_all_core_loss": 12.532559653690884, "Trainer/where": 0.1742857142857143, "Trainer/epoch": 6, "Trainer/steps_train": 245}
+{"Losses/train_all_loss": 12.028561905452184, "Losses/train_all_loss_mask": 0.003908876593257966, "Losses/train_all_loss_dice": 9.698760529926846, "Losses/train_all_loss_iou": 0.10878030185870427, "Losses/train_all_loss_class": 2.1428432097658514, "Losses/train_all_core_loss": 12.028561905452184, "Trainer/where": 0.1992857142857143, "Trainer/epoch": 7, "Trainer/steps_train": 280}
+{"Losses/train_all_loss": 12.847667748587472, "Losses/train_all_loss_mask": 0.002982073316317318, "Losses/train_all_loss_dice": 11.120562178747994, "Losses/train_all_loss_iou": 0.07551417656109802, "Losses/train_all_loss_class": 1.5919497845993777, "Losses/train_all_core_loss": 12.847667748587472, "Trainer/where": 0.22428571428571428, "Trainer/epoch": 8, "Trainer/steps_train": 315}
+{"Losses/train_all_loss": 11.098733384268625, "Losses/train_all_loss_mask": 0.007828620975903635, "Losses/train_all_loss_dice": 9.035280391148158, "Losses/train_all_loss_iou": 0.10779441692700077, "Losses/train_all_loss_class": 1.7990862643612282, "Losses/train_all_core_loss": 11.098733384268625, "Trainer/where": 0.24928571428571428, "Trainer/epoch": 9, "Trainer/steps_train": 350}
+{"Losses/train_all_loss": 10.939446571895054, "Losses/train_all_loss_mask": 0.003415139138399224, "Losses/train_all_loss_dice": 9.397412940434046, "Losses/train_all_loss_iou": 0.050908478014337434, "Losses/train_all_loss_class": 1.4228219677827187, "Losses/train_all_core_loss": 10.939446571895054, "Trainer/where": 0.2742857142857143, "Trainer/epoch": 10, "Trainer/steps_train": 385}
+{"Losses/train_all_loss": 11.260014452253069, "Losses/train_all_loss_mask": 0.0037810460206986005, "Losses/train_all_loss_dice": 9.496538509641375, "Losses/train_all_loss_iou": 0.06279493517223662, "Losses/train_all_loss_class": 1.6250600540411793, "Losses/train_all_core_loss": 11.260014452253069, "Trainer/where": 0.29928571428571427, "Trainer/epoch": 11, "Trainer/steps_train": 420}
+{"Losses/train_all_loss": 12.587461267198835, "Losses/train_all_loss_mask": 0.005486930625712765, "Losses/train_all_loss_dice": 10.807248279026576, "Losses/train_all_loss_iou": 0.12015448743186426, "Losses/train_all_loss_class": 1.550319859199226, "Losses/train_all_core_loss": 12.587461267198835, "Trainer/where": 0.3242857142857143, "Trainer/epoch": 12, "Trainer/steps_train": 455}
+{"Losses/train_all_loss": 12.622871085575648, "Losses/train_all_loss_mask": 0.0038172564951569907, "Losses/train_all_loss_dice": 11.162222140175956, "Losses/train_all_loss_iou": 0.05127683051479315, "Losses/train_all_loss_class": 1.3330271471424826, "Losses/train_all_core_loss": 12.622871085575648, "Trainer/where": 0.3492857142857143, "Trainer/epoch": 13, "Trainer/steps_train": 490}
+{"Losses/train_all_loss": 10.91449010031564, "Losses/train_all_loss_mask": 0.0021623846517676223, "Losses/train_all_loss_dice": 9.109422874450683, "Losses/train_all_loss_iou": 0.03527758292870463, "Losses/train_all_loss_class": 1.7265419553599453, "Losses/train_all_core_loss": 10.91449010031564, "Trainer/where": 0.3742857142857143, "Trainer/epoch": 14, "Trainer/steps_train": 525}
+{"Losses/train_all_loss": 12.423973451341901, "Losses/train_all_loss_mask": 0.002862129371247387, "Losses/train_all_loss_dice": 10.686428315298897, "Losses/train_all_loss_iou": 0.07063553876796505, "Losses/train_all_loss_class": 1.6096669961332477, "Losses/train_all_core_loss": 12.423973451341901, "Trainer/where": 0.3992857142857143, "Trainer/epoch": 15, "Trainer/steps_train": 560}
+{"Losses/train_all_loss": 13.769169889177595, "Losses/train_all_loss_mask": 0.004980977749385472, "Losses/train_all_loss_dice": 12.173356628417968, "Losses/train_all_loss_iou": 0.042235918748857716, "Losses/train_all_loss_class": 1.4539576956204006, "Losses/train_all_core_loss": 13.769169889177595, "Trainer/where": 0.42428571428571427, "Trainer/epoch": 16, "Trainer/steps_train": 595}
+{"Losses/train_all_loss": 11.263181223188127, "Losses/train_all_loss_mask": 0.0019078197024230447, "Losses/train_all_loss_dice": 9.721550485066006, "Losses/train_all_loss_iou": 0.0294029886584862, "Losses/train_all_loss_class": 1.4740716303233057, "Losses/train_all_core_loss": 11.263181223188127, "Trainer/where": 0.4492857142857143, "Trainer/epoch": 17, "Trainer/steps_train": 630}
+{"Losses/train_all_loss": 12.981634317125593, "Losses/train_all_loss_mask": 0.010148391163342499, "Losses/train_all_loss_dice": 11.691525718144009, "Losses/train_all_loss_iou": 0.06949597106098995, "Losses/train_all_loss_class": 1.0176445866269725, "Losses/train_all_core_loss": 12.981634317125593, "Trainer/where": 0.4742857142857143, "Trainer/epoch": 18, "Trainer/steps_train": 665}
+{"Losses/train_all_loss": 11.267893355233328, "Losses/train_all_loss_mask": 0.0026320721621492077, "Losses/train_all_loss_dice": 9.648453426361083, "Losses/train_all_loss_iou": 0.06427928070271653, "Losses/train_all_loss_class": 1.502519281820527, "Losses/train_all_core_loss": 11.267893355233328, "Trainer/where": 0.4992857142857143, "Trainer/epoch": 19, "Trainer/steps_train": 700}
+{"Losses/train_all_loss": 12.060971723284041, "Losses/train_all_loss_mask": 0.004294625139196536, "Losses/train_all_loss_dice": 10.43238205228533, "Losses/train_all_loss_iou": 0.0722124845030651, "Losses/train_all_loss_class": 1.470484510040842, "Losses/train_all_core_loss": 12.060971723284041, "Trainer/where": 0.5242857142857142, "Trainer/epoch": 20, "Trainer/steps_train": 735}
+{"Losses/train_all_loss": 10.979690497262137, "Losses/train_all_loss_mask": 0.002396076506036999, "Losses/train_all_loss_dice": 9.051968785694667, "Losses/train_all_loss_iou": 0.03635571563832595, "Losses/train_all_loss_class": 1.8434445864254876, "Losses/train_all_core_loss": 10.979690497262137, "Trainer/where": 0.5492857142857143, "Trainer/epoch": 21, "Trainer/steps_train": 770}
+{"Losses/train_all_loss": 10.912646198272705, "Losses/train_all_loss_mask": 0.002020072060570653, "Losses/train_all_loss_dice": 9.306321804864066, "Losses/train_all_loss_iou": 0.03728623108545435, "Losses/train_all_loss_class": 1.5286366003632013, "Losses/train_all_core_loss": 10.912646198272705, "Trainer/where": 0.5742857142857143, "Trainer/epoch": 22, "Trainer/steps_train": 805}
+{"Losses/train_all_loss": 12.972393226623534, "Losses/train_all_loss_mask": 0.0029303550138138233, "Losses/train_all_loss_dice": 11.663762542179652, "Losses/train_all_loss_iou": 0.03950240197342022, "Losses/train_all_loss_class": 1.210520985183705, "Losses/train_all_core_loss": 12.972393226623534, "Trainer/where": 0.5992857142857143, "Trainer/epoch": 23, "Trainer/steps_train": 840}
+{"Losses/train_all_loss": 11.836965492793492, "Losses/train_all_loss_mask": 0.002555759671875941, "Losses/train_all_loss_dice": 9.82940263066973, "Losses/train_all_loss_iou": 0.06007927144362059, "Losses/train_all_loss_class": 1.8963684562028253, "Losses/train_all_core_loss": 11.836965492793492, "Trainer/where": 0.6242857142857143, "Trainer/epoch": 24, "Trainer/steps_train": 875}
+{"Losses/train_all_loss": 12.333917222704207, "Losses/train_all_loss_mask": 0.005893107972640012, "Losses/train_all_loss_dice": 10.638660866873606, "Losses/train_all_loss_iou": 0.1051822858673404, "Losses/train_all_loss_class": 1.4722119661713284, "Losses/train_all_core_loss": 12.333917222704207, "Trainer/where": 0.6492857142857142, "Trainer/epoch": 25, "Trainer/steps_train": 910}
+{"Losses/train_all_loss": 12.976327841622489, "Losses/train_all_loss_mask": 0.0058499143504637426, "Losses/train_all_loss_dice": 11.136853994641985, "Losses/train_all_loss_iou": 0.07329373683772117, "Losses/train_all_loss_class": 1.6491819693573884, "Losses/train_all_core_loss": 12.976327841622489, "Trainer/where": 0.6742857142857143, "Trainer/epoch": 26, "Trainer/steps_train": 945}
+{"Losses/train_all_loss": 12.715955025809151, "Losses/train_all_loss_mask": 0.0051494090393784326, "Losses/train_all_loss_dice": 11.038715130942208, "Losses/train_all_loss_iou": 0.07952312049164903, "Losses/train_all_loss_class": 1.494728443427344, "Losses/train_all_core_loss": 12.715955025809151, "Trainer/where": 0.6992857142857143, "Trainer/epoch": 27, "Trainer/steps_train": 980}
+{"Losses/train_all_loss": 12.254196275983539, "Losses/train_all_loss_mask": 0.002468374198568719, "Losses/train_all_loss_dice": 10.371674939564295, "Losses/train_all_loss_iou": 0.035466401658517756, "Losses/train_all_loss_class": 1.7976875406590158, "Losses/train_all_core_loss": 12.254196275983539, "Trainer/where": 0.7242857142857143, "Trainer/epoch": 28, "Trainer/steps_train": 1015}
+{"Losses/train_all_loss": 11.89058301108224, "Losses/train_all_loss_mask": 0.0033038316294550897, "Losses/train_all_loss_dice": 10.133634703499931, "Losses/train_all_loss_iou": 0.06524576840498152, "Losses/train_all_loss_class": 1.6256259542756848, "Losses/train_all_core_loss": 11.89058301108224, "Trainer/where": 0.7492857142857143, "Trainer/epoch": 29, "Trainer/steps_train": 1050}
+{"Losses/train_all_loss": 12.994428975241524, "Losses/train_all_loss_mask": 0.0032894188288732297, "Losses/train_all_loss_dice": 11.455551188332693, "Losses/train_all_loss_iou": 0.05269411236804444, "Losses/train_all_loss_class": 1.4203952280871037, "Losses/train_all_core_loss": 12.994428975241524, "Trainer/where": 0.7742857142857142, "Trainer/epoch": 30, "Trainer/steps_train": 1085}
+{"Losses/train_all_loss": 11.028971740177699, "Losses/train_all_loss_mask": 0.0017663538056824888, "Losses/train_all_loss_dice": 9.26269657952445, "Losses/train_all_loss_iou": 0.020907437572480245, "Losses/train_all_loss_class": 1.7100406129179255, "Losses/train_all_core_loss": 11.028971740177699, "Trainer/where": 0.7992857142857143, "Trainer/epoch": 31, "Trainer/steps_train": 1120}
+{"Losses/train_all_loss": 12.77100260598319, "Losses/train_all_loss_mask": 0.005347782664466649, "Losses/train_all_loss_dice": 11.073454679761614, "Losses/train_all_loss_iou": 0.1097778601632204, "Losses/train_all_loss_class": 1.480814137798734, "Losses/train_all_core_loss": 12.77100260598319, "Trainer/where": 0.8242857142857142, "Trainer/epoch": 32, "Trainer/steps_train": 1155}
+{"Losses/train_all_loss": 12.876578099387032, "Losses/train_all_loss_mask": 0.014166244934313, "Losses/train_all_loss_dice": 10.838990374973841, "Losses/train_all_loss_iou": 0.09881567690504849, "Losses/train_all_loss_class": 1.6554468802575555, "Losses/train_all_core_loss": 12.876578099387032, "Trainer/where": 0.8492857142857142, "Trainer/epoch": 33, "Trainer/steps_train": 1190}
+{"Losses/train_all_loss": 10.704918697902135, "Losses/train_all_loss_mask": 0.0019853250996675342, "Losses/train_all_loss_dice": 8.530561372212, "Losses/train_all_loss_iou": 0.022953493090088678, "Losses/train_all_loss_class": 2.1116973195324786, "Losses/train_all_core_loss": 10.704918697902135, "Trainer/where": 0.8742857142857142, "Trainer/epoch": 34, "Trainer/steps_train": 1225}
+{"Losses/train_all_loss": 12.674189213344029, "Losses/train_all_loss_mask": 0.004104748995242906, "Losses/train_all_loss_dice": 11.569254003252302, "Losses/train_all_loss_iou": 0.013460466006654315, "Losses/train_all_loss_class": 1.0093798162881285, "Losses/train_all_core_loss": 12.674189213344029, "Trainer/where": 0.8992857142857142, "Trainer/epoch": 35, "Trainer/steps_train": 1260}
+{"Losses/train_all_loss": 13.152547141483852, "Losses/train_all_loss_mask": 0.00731085481404859, "Losses/train_all_loss_dice": 11.433105080468314, "Losses/train_all_loss_iou": 0.04558718047472732, "Losses/train_all_loss_class": 1.5276377756015531, "Losses/train_all_core_loss": 13.152547141483852, "Trainer/where": 0.9242857142857142, "Trainer/epoch": 36, "Trainer/steps_train": 1295}
+{"Losses/train_all_loss": 10.936215046473912, "Losses/train_all_loss_mask": 0.00343067699328198, "Losses/train_all_loss_dice": 9.400099325180054, "Losses/train_all_loss_iou": 0.0718443956205322, "Losses/train_all_loss_class": 1.3956576879734972, "Losses/train_all_core_loss": 10.936215046473912, "Trainer/where": 0.9492857142857142, "Trainer/epoch": 37, "Trainer/steps_train": 1330}
+{"Losses/train_all_loss": 13.038009902409145, "Losses/train_all_loss_mask": 0.001936620579467022, "Losses/train_all_loss_dice": 11.594574492318289, "Losses/train_all_loss_iou": 0.04597411672626289, "Losses/train_all_loss_class": 1.3587288303616722, "Losses/train_all_core_loss": 13.038009902409145, "Trainer/where": 0.9742857142857142, "Trainer/epoch": 38, "Trainer/steps_train": 1365}
+{"Losses/train_all_loss": 12.413451303754535, "Losses/train_all_loss_mask": 0.003644983982667327, "Losses/train_all_loss_dice": 10.79392305782863, "Losses/train_all_loss_iou": 0.06104640555193847, "Losses/train_all_loss_class": 1.4855821575769887, "Losses/train_all_core_loss": 12.413451303754535, "Trainer/where": 0.9992857142857142, "Trainer/epoch": 39, "Trainer/steps_train": 1400}

tensorboard/events.out.tfevents.1736037893.creative-turing-2.3136573.0928bfd69-289f-40fa-b70d-977964f79bac

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

tensorboard/events.out.tfevents.1736039984.creative-turing-2.3143541.09b0a183e-0877-483b-8228-cdb52183d13b

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

tensorboard/events.out.tfevents.1736040155.creative-turing-2.3145589.0dbed95eb-8481-4404-acdf-20879f1218f7

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

tensorboard/events.out.tfevents.1737390155.creative-turing-2.1898139.03b45d984-ad8b-4ade-ab73-ec2387852ef3

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