Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,15 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+assets/architecture.png filter=lfs diff=lfs merge=lfs -text
+assets/data_1.png filter=lfs diff=lfs merge=lfs -text
+assets/data_2.png filter=lfs diff=lfs merge=lfs -text
+assets/data_3.png filter=lfs diff=lfs merge=lfs -text
+assets/data_4.png filter=lfs diff=lfs merge=lfs -text
+assets/data_5.png filter=lfs diff=lfs merge=lfs -text
+assets/data_6.png filter=lfs diff=lfs merge=lfs -text
+assets/finetune_detail.png filter=lfs diff=lfs merge=lfs -text
+assets/teasser.png filter=lfs diff=lfs merge=lfs -text
+assets/training_metrics.png filter=lfs diff=lfs merge=lfs -text
+assets/val_sample_early.jpg filter=lfs diff=lfs merge=lfs -text
+assets/val_sample_final.jpg filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,22 @@
+# Python
+__pycache__/
+*.py[cod]
+*.egg-info/
+*.egg
+
+# IDE
+.vscode/
+.idea/
+
+# OS
+.DS_Store
+Thumbs.db
+
+# Temporary files
+*.tmp
+*.bak
+
+# Training artifacts (not needed on HF)
+wandb/
+runs/
+*.log

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Adithya N Raj
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,3 +1,369 @@
----
-license: mit
----
+---
+language: en
+license: mit
+tags:
+  - court-detection
+  - keypoint-detection
+  - sports-analysis
+  - badminton
+  - tennis
+  - pytorch
+  - computer-vision
+  - octave-convolution
+  - geometric-constraints
+  - polar-transform
+datasets:
+  - adithyanraj03/CourtKeyNet-Dataset
+metrics:
+  - pck
+  - iou
+pipeline_tag: keypoint-detection
+library_name: pytorch
+model-index:
+  - name: CourtKeyNet
+    results:
+      - task:
+          type: keypoint-detection
+          name: Court Keypoint Detection
+        dataset:
+          name: CourtKeyNet Badminton Court Dataset
+          type: custom
+        metrics:
+          - name: PCK
+            type: pck
+            value: 99.99
+          - name: IoU
+            type: iou
+            value: 99.67
+---
+
+<div align="center">
+  <img src="assets/teasser.png" alt="CourtKeyNet Banner" width="100%">
+
+# CourtKeyNet: A Novel Octave-Based Architecture for Precision Court Detection
+
+**Adithya N Raj** 
+
+[![Paper](https://img.shields.io/static/v1?label=Paper&message=ScienceDirect&color=red&logo=elsevier&logoColor=white)](https://www.sciencedirect.com/science/article/pii/S2666827026000496)
+[![GitHub](https://img.shields.io/static/v1?label=Code&message=GitHub&color=black&logo=github)](https://github.com/adithyanraj03/CourtKeyNet)
+[![Dataset](https://img.shields.io/badge/%F0%9F%93%81%20Dataset-GitHub-blue)](https://github.com/adithyanraj03/Paper_09_Data-Set_CourtKeyNet)
+[![License](https://img.shields.io/badge/License-MIT-green)](LICENSE)
+[![SafeTensors](https://img.shields.io/badge/Format-SafeTensors-8A2BE2)](https://huggingface.co/docs/safetensors)
+
+</div>
+
+---
+
+## Model Summary
+
+**CourtKeyNet** is a lightweight deep learning architecture for precision court keypoint detection in sports videos. It detects 4 corner keypoints of badminton and tennis courts from a single image, enabling downstream applications like court homography estimation, match statistics generation, and automated broadcasting systems.
+
+Published in [Machine Learning with Applications (Elsevier)](https://www.sciencedirect.com/science/article/pii/S2666827026000496), 2026.
+
+### Key Features
+
+- 🎯 **99.99% PCK** and **99.67% IoU** on the fine-tuned model
+- ⚡ **Lightweight and fast** — efficient architecture for real-time use
+- 🔬 **Octave Feature Extractor** — Multi-frequency feature decomposition for capturing fine court details and global structural context
+- 🎱 **Polar Transform Attention** — Boundary detection in polar coordinates for precise court line localization
+- 📐 **Geometric Consistency Loss** — Ensures structurally valid quadrilateral outputs
+- 🏸 Supports both **badminton** and **tennis** court detection
+
+---
+
+## Model Variants
+
+| Variant | File | Format | Val Loss | PCK (%) | IoU (%) | Description |
+|---------|------|--------|----------|---------|---------|-------------|
+| **CourtKeyNet-Base** | `pretrained/courtkeynet_base.safetensors` | SafeTensors | 0.0977 | 99.09 | 94.33 | Pre-trained on 140k+ images from scratch |
+| **CourtKeyNet-Finetuned** ⭐ | `finetuned/courtkeynet_finetuned.safetensors` | SafeTensors | 0.0013 | 99.99 | 99.67 | Fine-tuned on 7k+ clean annotated images |
+
+> **Recommendation:** Use the **fine-tuned** variant for best results. The pre-trained base model is provided for researchers who wish to fine-tune on their own custom court datasets.
+
+---
+
+## Quick Start
+
+### 1. Download Weights
+
+```bash
+# Install huggingface CLI
+pip install "huggingface_hub[cli]"
+
+# Download the full repository
+huggingface-cli download Cracked-ANJ/CourtKeyNet --local-dir ./courtkeynet-weights
+
+# Or download only the fine-tuned model (recommended)
+huggingface-cli download Cracked-ANJ/CourtKeyNet finetuned/courtkeynet_finetuned.safetensors --local-dir ./courtkeynet-weights
+```
+
+### 2. Run Inference
+
+For inference, training, and fine-tuning scripts, use the full source code from the **[GitHub Repository](https://github.com/adithyanraj03/CourtKeyNet)**:
+
+```bash
+# Clone the source code
+git clone https://github.com/adithyanraj03/CourtKeyNet.git
+cd CourtKeyNet
+
+# Install dependencies
+pip install -r requirements.txt
+
+# Run Inference Studio (GUI)
+cd courtkeynet
+python inference.py
+```
+
+> The Inference Studio provides a complete GUI with image/video/webcam support, confidence scoring, and real-time visualization.
+
+---
+
+## Architecture
+
+<div align="center">
+  <img src="assets/architecture.png" alt="CourtKeyNet Architecture" width="80%">
+</div>
+
+
+
+<details>
+<summary style="font-size: 1.2em; font-weight: bold; cursor: pointer;">🔍 How Confidence Detection Works (Visual Explanation)</summary>
+
+### How Confidence Detection Works
+
+The model (`CourtKeyNet`) works like this:
+
+![Architecture](assets/data_1.png)
+
+**Problem**: It has no "court detector" — it assumes every image IS a court!
+
+---
+
+#### What the Model Actually Outputs Internally
+
+When you run `model(image)`, it returns a dictionary with these components:
+
+```python
+outputs = {
+    'heatmaps': Tensor[B, 4, 160, 160],      # 4 gaussian peaks (one per corner)
+    'kpts_init': Tensor[B, 4, 2],            # Initial keypoints from heatmaps
+    'kpts_refined': Tensor[B, 4, 2],         # Final refined keypoints
+    'features': Tensor[B, 256, 20, 20]       # Feature maps (optional)
+}
+```
+
+#### Visualization of Heatmap Output
+
+For a **real court image**:
+```text
+Heatmap for Corner 0 (Top-Left):
+```
+![Heatmap](assets/data_2.png)
+
+For a **non-court image** (e.g., random person):
+```text
+Heatmap for Corner 0:
+```
+![Heatmap](assets/data_3.png)
+
+---
+
+#### 3 Confidence Metrics
+
+##### 1️⃣ **Heatmap Peak Confidence** (Primary Signal)
+
+**What it measures**: How "peaky" the heatmap is
+
+```python
+max_values = heatmaps.max(dim=(2,3))  # Find highest value in each heatmap
+conf_heatmap = max_values.mean()      # Average across 4 corners
+```
+
+**Visual comparison**:
+
+![Comparison](assets/data_4.png)
+
+---
+
+##### 2️⃣ **Heatmap Entropy** (Uncertainty)
+
+**What it measures**: How "spread out" the probability is
+
+```python
+# Entropy = -Σ(p * log(p))
+# Low entropy = focused (good)
+# High entropy = random noise (bad)
+```
+
+**Visual comparison**:
+
+![Entropy_Comparison](assets/data_5.png)
+
+---
+
+##### 3️⃣ **Geometric Validity** (Shape Check)
+
+**What it checks**: Does the quad look like a real court?
+
+```text
+Checklist:
+✓ Are corners in correct positions? (TL upper-left, BR lower-right)
+✓ Is the quad convex? (no crossed lines)
+✓ Is the area reasonable? (not too tiny, not entire image)
+✓ Is aspect ratio court-like? (not a thin line)
+```
+
+**Visual examples**:
+
+![Visual_Example](assets/data_6.png)
+
+</details>
+
+---
+
+## Training Details
+
+### Pre-training (from scratch)
+
+| Parameter | Value |
+|-----------|-------|
+| **Dataset** | 140,000+ badminton & tennis court images |
+| **Epochs** | 158 (early stopping, patience=20) |
+| **Best Epoch** | 138 |
+| **Optimizer** | AdamW |
+| **Learning Rate** | 5e-5 (cosine schedule) |
+| **Batch Size** | 47 |
+| **Image Size** | 640×640 |
+| **Mixed Precision** | ✓ FP16 |
+| **Hardware** | NVIDIA RTX 5090 (32GB) |
+
+### Fine-tuning
+
+| Parameter | Value |
+|-----------|-------|
+| **Base Model** | Pre-trained CourtKeyNet-Base (epoch 138) |
+| **Dataset** | 7,000+ precisely annotated clean images |
+| **Epochs** | 79 (early stopping, patience=20) |
+| **Best Epoch** | 59 |
+| **Learning Rate** | 1e-4 |
+| **Batch Size** | 48 |
+| **Geometric Loss** | Enabled (edge + diagonal + angle constraints) |
+
+---
+
+## Evaluation Results
+
+### Fine-tuned Model (Recommended)
+
+| Metric | Score |
+|--------|-------|
+| **PCK** (Percentage of Correct Keypoints) | **99.99%** |
+| **IoU** (Intersection over Union) | **99.67%** |
+| **Validation Loss** | **0.0013** |
+
+### Pre-trained Base Model
+
+| Metric | Score |
+|--------|-------|
+| **PCK** | 99.09% |
+| **IoU** | 94.33% |
+| **Validation Loss** | 0.0977 |
+
+### Training Metrics
+
+<div align="center">
+  <img src="assets/training_metrics.png" alt="Training Metrics" width="100%">
+  <p><em>Pre-training metrics across 158 epochs showing loss convergence and PCK/IoU progression</em></p>
+</div>
+
+<div align="center">
+  <img src="assets/finetune_detail.png" alt="Fine-tuning Detail" width="100%">
+  <p><em>Fine-tuning convergence: rapid loss decrease with PCK reaching 99.99% and IoU reaching 99.67%</em></p>
+</div>
+
+### Validation Samples
+
+<div align="center">
+  <table>
+    <tr>
+      <td align="center"><strong>Epoch 1 (Early Fine-tuning)</strong></td>
+      <td align="center"><strong>Epoch 79 (Final)</strong></td>
+    </tr>
+    <tr>
+      <td><img src="assets/val_sample_early.jpg" alt="Early Validation" width="100%"></td>
+      <td><img src="assets/val_sample_final.jpg" alt="Final Validation" width="100%"></td>
+    </tr>
+  </table>
+  <p><em>Red = Predicted court boundary | Green = Ground truth | Yellow dots = Keypoints</em></p>
+</div>
+
+---
+
+## Fine-tuning on Custom Data
+
+To fine-tune CourtKeyNet on your own court dataset (any sport):
+
+```bash
+# 1. Clone the source code
+git clone https://github.com/adithyanraj03/CourtKeyNet.git
+cd CourtKeyNet
+
+# 2. Download weights
+huggingface-cli download Cracked-ANJ/CourtKeyNet pretrained/courtkeynet_base.pt --local-dir ./weights
+
+# 3. Fine-tune
+cd courtkeynet
+python finetune.py
+```
+
+> **Note:** Fine-tuning requires only **5,000–7,000** clean annotated images. Training from scratch requires **140,000+** images.
+
+---
+
+## Intended Use
+
+### ✅ Appropriate Uses
+- Badminton and tennis court detection in sports videos
+- Court homography estimation for tactical analysis
+- Generating match statistics from broadcast footage
+- Automated camera calibration for sports broadcasting
+- Academic research in sports video analysis
+
+### ⚠️ Limitations
+- Trained specifically on badminton and tennis courts; other court types may require fine-tuning
+- Expects single-court images; multi-court scenes are not supported
+- Performance may degrade on heavily occluded courts or extreme camera angles
+- The model detects 4 outer corners only (not inner court lines)
+
+---
+
+## Citation
+
+If you use CourtKeyNet in your research, please cite:
+
+**Paper:** [CourtKeyNet: A novel octave-based architecture for precision badminton court detection with geometric constraints](https://www.sciencedirect.com/science/article/pii/S2666827026000496)  
+**DOI:** [10.1016/j.mlwa.2026.100884](https://doi.org/10.1016/j.mlwa.2026.100884)
+
+```bibtex
+@article{NRAJ2026100884,
+  title     = {CourtKeyNet: A novel octave-based architecture for precision badminton court detection with geometric constraints},
+  journal   = {Machine Learning with Applications},
+  volume    = {24},
+  pages     = {100884},
+  year      = {2026},
+  issn      = {2666-8270},
+  doi       = {https://doi.org/10.1016/j.mlwa.2026.100884},
+  url       = {https://www.sciencedirect.com/science/article/pii/S2666827026000496},
+  author    = {Adithya N Raj and Prethija G.}
+}
+```
+
+---
+
+## License
+
+This project is released under the [MIT License](LICENSE), suitable for both academic and commercial use.
+
+## Contact
+
+For questions or collaboration opportunities: **adithyanraj03@gmail.com**

config.yaml

@@ -0,0 +1,93 @@
+# Model Configuration (Manuscript Section 4)
+model:
+  name: "CourtKeyNet"
+  
+  # Octave Feature Extractor
+  ofe:
+    channels_per_band: 64  # C for high/mid/low paths
+    stem_channels: 64
+    
+  # Polar Transform Attention  
+  pta:
+    enabled: true
+    radial_bins: 64      # Manuscript says fine-grained polar grid
+    angular_bins: 128
+    
+  # Keypoint Localization
+  heatmap_sigma: 8.0     # Increased from 2.0 to help convergence
+  num_keypoints: 4
+  feature_dim: 128
+  
+  # Transformer Refinement
+  transformer:
+    num_layers: 2
+    num_heads: 4
+    dim_feedforward: 512
+    dropout: 0.1
+    
+  # Quadrilateral Constraint Module
+  qcm:
+    hidden_dims: [64, 128]
+    output_dim: 128
+
+# Training Configuration
+train:
+  epochs: 300
+  batch_size: 47
+  num_workers: 4
+  imgsz: 640
+  early_stopping_patience: 20  # Stop if no improvement for N epochs
+  
+  # Optimizer (Manuscript: AdamW)
+  optimizer: "adamw"
+  lr0: 0.00005 # Reduced further to prevent divergence with high loss weights
+  lrf: 0.01  # Final LR multiplier
+  weight_decay: 0.0005
+  momentum: 0.937
+  
+  # LR Scheduler
+  warmup_epochs: 5
+  scheduler: "cosine"
+  
+  # Loss Weights (Manuscript Eq. 47)
+  # Geometric losses disabled initially (set to 0) to let model learn keypoints first
+  loss_weights:
+    keypoint: 10.0     # Primary loss for coordinate accuracy
+    heatmap: 20.0      # Heatmap supervision
+    edge: 0.0          # Disabled initially - enable after model converges
+    diagonal: 0.0      # Disabled initially
+    angle: 0.0         # Disabled initially
+  
+  # Geometric loss warmup (linearly scale geo losses over N epochs)
+  # Only relevant when edge/diagonal/angle > 0
+  # geo_warmup_epochs: 10 #enabled in finetune only
+  
+  # Training Tricks
+  mixed_precision: true
+  grad_clip: 1.0
+  ema_decay: 0.9999
+  
+  # Checkpointing
+  save_interval: 5
+  save_best: true
+  
+  # Paths
+  project: "runs/courtkeynet"
+  name: "exp"
+
+# Validation
+val:
+  batch_size: 32
+  interval: 1  # Validate every N epochs
+
+# Weights & Biases Configuration
+wandb:
+  enabled: true
+  project: "CourtKeyNet"
+  entity: null  # Set to your wandb username or team name, or leave null for default
+  name: CourtKeyNet    # Run name (auto-generated if null)
+  tags: ["badminton", "court-detection", "keypoint"]
+  log_freq: 100  # Log every N batches
+  
+# Device
+device: "cuda"  # or "cpu"

finetuned/courtkeynet_finetuned.safetensors

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

finetuned/training_log.txt

@@ -0,0 +1,785 @@
+(gpu) PS C:\Research\CourtKeyNet> python .\finetune.py
+============================================================
+  CourtKeyNet Fine-tuning Script
+  Clean Dataset: 7k+ perfectly annotated images
+============================================================
+
+Selected pretrained weights: C:/Research/CourtKeyNet/runs/courtkeynet/exp_20260121_223500_wijqjzd9/best.pt
+Clean dataset: C:\Research\Datasets\Badminton_court\7k+Clean_Dataset
+Using device: cuda
+wandb: Currently logged in as: adithyanraj03 to https://api.wandb.ai. Use `wandb login --relogin` to force relogin
+wandb: Tracking run with wandb version 0.23.1
+wandb: Run data is saved locally in C:\Research\CourtKeyNet\wandb\run-20260129_061557-0fnvysj4
+wandb: Run `wandb offline` to turn off syncing.
+wandb: Syncing run Finetune_01/29_06:15
+wandb:  View project at https://wandb.ai/adithyanraj03/courtkeynet
+wandb:  View run at https://wandb.ai/adithyanraj03/courtkeynet/runs/0fnvysj4
+wandb: Detected [huggingface_hub.inference] in use.
+wandb: Use W&B Weave for improved LLM call tracing. Install Weave with `pip install weave` then add `import weave` to the top of your script.
+wandb: For more information, check out the docs at: https://weave-docs.wandb.ai/
+Weights & Biases initialized: https://wandb.ai/adithyanraj03/courtkeynet/runs/0fnvysj4
+[train] Found 58123 images
+[valid] Found 16826 images
+[finetune-train] Found 58123 images
+[finetune-valid] Found 16826 images
+
+Loading pretrained weights...
+  Loaded from epoch 138, best_val_loss=0.09765742771935054
+Model parameters: 1,238,856
+Fine-tuning LR: 0.0001
+Checkpoints will be saved to: runs\courtkeynet_finetune\finetune_20260129_061601_0fnvysj4
+
+============================================================
+  Starting Fine-tuning...
+============================================================
+
+Epoch 0: 100%|████████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [20:27<00:00,  1.01s/it, loss=0.0173] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.36it/s] 
+
+Epoch 1/100
+  Train Loss: 0.0586
+  Val Loss: 0.0181
+  Val PCK: 0.9943
+  Val IoU: 0.9833
+  ✓ Saved best model (val_loss=0.0181)
+Epoch 1: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:02<00:00,  2.01it/s, loss=0.00963]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.38it/s] 
+
+Epoch 2/100
+  Train Loss: 0.0132
+  Val Loss: 0.0098
+  Val PCK: 0.9976
+  Val IoU: 0.9901
+  ✓ Saved best model (val_loss=0.0098)
+Epoch 2: 100%|████████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.01it/s, loss=0.0109]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.38it/s] 
+
+Epoch 3/100
+  Train Loss: 0.0106
+  Val Loss: 0.0117
+  Val PCK: 0.9964
+  Val IoU: 0.9896
+  No improvement for 1/20 epochs
+Epoch 3: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:02<00:00,  2.01it/s, loss=0.00971] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.38it/s] 
+
+Epoch 4/100
+  Train Loss: 0.0087
+  Val Loss: 0.0076
+  Val PCK: 0.9983
+  Val IoU: 0.9924
+  ✓ Saved best model (val_loss=0.0076)
+Epoch 4: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:01<00:00,  2.01it/s, loss=0.00753]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.39it/s] 
+
+Epoch 5/100
+  Train Loss: 0.0084
+  Val Loss: 0.0066
+  Val PCK: 0.9983
+  Val IoU: 0.9932
+  ✓ Saved best model (val_loss=0.0066)
+Epoch 5: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [31:39<00:00,  1.57s/it, loss=0.00898]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:15<00:00,  1.40it/s] 
+
+Epoch 6/100
+  Train Loss: 0.0069
+  Val Loss: 0.0060
+  Val PCK: 0.9982
+  Val IoU: 0.9933
+  ✓ Saved best model (val_loss=0.0060)
+Epoch 6: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [32:04<00:00,  1.59s/it, loss=0.00831]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:21<00:00,  1.38it/s] 
+
+Epoch 7/100
+  Train Loss: 0.0070
+  Val Loss: 0.0057
+  Val PCK: 0.9986
+  Val IoU: 0.9937
+  ✓ Saved best model (val_loss=0.0057)
+Epoch 7: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [18:15<00:00,  1.10it/s, loss=0.00608] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.40it/s] 
+
+Epoch 8/100
+  Train Loss: 0.0065
+  Val Loss: 0.0064
+  Val PCK: 0.9980
+  Val IoU: 0.9936
+  No improvement for 1/20 epochs
+Epoch 8: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [11:27<00:00,  1.76it/s, loss=0.00956] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.39it/s] 
+
+Epoch 9/100
+  Train Loss: 0.0069
+  Val Loss: 0.0069
+  Val PCK: 0.9985
+  Val IoU: 0.9930
+  No improvement for 2/20 epochs
+Epoch 9: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:01<00:00,  2.01it/s, loss=0.00876] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.39it/s] 
+
+Epoch 10/100
+  Train Loss: 0.0062
+  Val Loss: 0.0076
+  Val PCK: 0.9982
+  Val IoU: 0.9932
+  No improvement for 3/20 epochs
+Epoch 10: 100%|█████████████████████████��████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.02it/s, loss=0.00628]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.38it/s] 
+
+Epoch 11/100
+  Train Loss: 0.0059
+  Val Loss: 0.0051
+  Val PCK: 0.9989
+  Val IoU: 0.9946
+  ✓ Saved best model (val_loss=0.0051)
+Epoch 11: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.01it/s, loss=0.0072]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.38it/s] 
+
+Epoch 12/100
+  Train Loss: 0.0064
+  Val Loss: 0.0048
+  Val PCK: 0.9988
+  Val IoU: 0.9950
+  ✓ Saved best model (val_loss=0.0048)
+Epoch 12: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:02<00:00,  2.01it/s, loss=0.00574]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.39it/s] 
+
+Epoch 13/100
+  Train Loss: 0.0057
+  Val Loss: 0.0053
+  Val PCK: 0.9989
+  Val IoU: 0.9946
+  No improvement for 1/20 epochs
+Epoch 13: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:01<00:00,  2.01it/s, loss=0.00631] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.38it/s] 
+
+Epoch 14/100
+  Train Loss: 0.0059
+  Val Loss: 0.0043
+  Val PCK: 0.9990
+  Val IoU: 0.9951
+  ✓ Saved best model (val_loss=0.0043)
+Epoch 14: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:01<00:00,  2.01it/s, loss=0.00544] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.37it/s] 
+
+Epoch 15/100
+  Train Loss: 0.0057
+  Val Loss: 0.0069
+  Val PCK: 0.9980
+  Val IoU: 0.9939
+  No improvement for 1/20 epochs
+Epoch 15: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:01<00:00,  2.01it/s, loss=0.00487] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.39it/s] 
+
+Epoch 16/100
+  Train Loss: 0.0061
+  Val Loss: 0.0047
+  Val PCK: 0.9990
+  Val IoU: 0.9949
+  No improvement for 2/20 epochs
+Epoch 16: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [09:59<00:00,  2.02it/s, loss=0.00493] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.38it/s] 
+
+Epoch 17/100
+  Train Loss: 0.0058
+  Val Loss: 0.0035
+  Val PCK: 0.9989
+  Val IoU: 0.9953
+  ✓ Saved best model (val_loss=0.0035)
+Epoch 17: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.02it/s, loss=0.00418]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.40it/s] 
+
+Epoch 18/100
+  Train Loss: 0.0051
+  Val Loss: 0.0032
+  Val PCK: 0.9991
+  Val IoU: 0.9954
+  ✓ Saved best model (val_loss=0.0032)
+Epoch 18: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:02<00:00,  2.01it/s, loss=0.00595] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.39it/s] 
+
+Epoch 19/100
+  Train Loss: 0.0048
+  Val Loss: 0.0040
+  Val PCK: 0.9994
+  Val IoU: 0.9954
+  No improvement for 1/20 epochs
+Epoch 19: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.01it/s, loss=0.00491] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.39it/s] 
+
+Epoch 20/100
+  Train Loss: 0.0046
+  Val Loss: 0.0030
+  Val PCK: 0.9994
+  Val IoU: 0.9955
+  ✓ Saved best model (val_loss=0.0030)
+Epoch 20: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:01<00:00,  2.01it/s, loss=0.00482]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.38it/s] 
+
+Epoch 21/100
+  Train Loss: 0.0047
+  Val Loss: 0.0041
+  Val PCK: 0.9989
+  Val IoU: 0.9950
+  No improvement for 1/20 epochs
+Epoch 21: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.02it/s, loss=0.00384] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.38it/s] 
+
+Epoch 22/100
+  Train Loss: 0.0046
+  Val Loss: 0.0047
+  Val PCK: 0.9987
+  Val IoU: 0.9946
+  No improvement for 2/20 epochs
+Epoch 22: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.01it/s, loss=0.00503] 
+Validation: 100%|██████████████████████████████████████���█████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.39it/s] 
+
+Epoch 23/100
+  Train Loss: 0.0048
+  Val Loss: 0.0049
+  Val PCK: 0.9986
+  Val IoU: 0.9946
+  No improvement for 3/20 epochs
+Epoch 23: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.02it/s, loss=0.00434] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.39it/s] 
+
+Epoch 24/100
+  Train Loss: 0.0049
+  Val Loss: 0.0026
+  Val PCK: 0.9995
+  Val IoU: 0.9958
+  ✓ Saved best model (val_loss=0.0026)
+Epoch 24: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:01<00:00,  2.01it/s, loss=0.00425]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.38it/s] 
+
+Epoch 25/100
+  Train Loss: 0.0042
+  Val Loss: 0.0029
+  Val PCK: 0.9993
+  Val IoU: 0.9957
+  No improvement for 1/20 epochs
+Epoch 25: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.01it/s, loss=0.00602] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.38it/s] 
+
+Epoch 26/100
+  Train Loss: 0.0053
+  Val Loss: 0.0035
+  Val PCK: 0.9995
+  Val IoU: 0.9955
+  No improvement for 2/20 epochs
+Epoch 26: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.01it/s, loss=0.00537] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.37it/s] 
+
+Epoch 27/100
+  Train Loss: 0.0044
+  Val Loss: 0.0026
+  Val PCK: 0.9994
+  Val IoU: 0.9958
+  ✓ Saved best model (val_loss=0.0026)
+Epoch 27: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.01it/s, loss=0.0028]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [01:59<00:00,  4.40it/s] 
+
+Epoch 28/100
+  Train Loss: 0.0040
+  Val Loss: 0.0025
+  Val PCK: 0.9994
+  Val IoU: 0.9960
+  ✓ Saved best model (val_loss=0.0025)
+Epoch 28: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:01<00:00,  2.01it/s, loss=0.00418]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████���| 526/526 [01:59<00:00,  4.39it/s] 
+
+Epoch 29/100
+  Train Loss: 0.0040
+  Val Loss: 0.0031
+  Val PCK: 0.9992
+  Val IoU: 0.9956
+  No improvement for 1/20 epochs
+Epoch 29: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.02it/s, loss=0.00418] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [05:13<00:00,  1.68it/s] 
+
+Epoch 30/100
+  Train Loss: 0.0052
+  Val Loss: 0.0031
+  Val PCK: 0.9991
+  Val IoU: 0.9958
+  No improvement for 2/20 epochs
+Epoch 30: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [32:20<00:00,  1.60s/it, loss=0.00388]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:18<00:00,  1.39it/s] 
+
+Epoch 31/100
+  Train Loss: 0.0040
+  Val Loss: 0.0020
+  Val PCK: 0.9994
+  Val IoU: 0.9962
+  ✓ Saved best model (val_loss=0.0020)
+Epoch 31: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [32:20<00:00,  1.60s/it, loss=0.0036]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:22<00:00,  1.38it/s] 
+
+Epoch 32/100
+  Train Loss: 0.0041
+  Val Loss: 0.0025
+  Val PCK: 0.9993
+  Val IoU: 0.9960
+  No improvement for 1/20 epochs
+Epoch 32: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [32:20<00:00,  1.60s/it, loss=0.00352] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:16<00:00,  1.40it/s] 
+
+Epoch 33/100
+  Train Loss: 0.0039
+  Val Loss: 0.0023
+  Val PCK: 0.9992
+  Val IoU: 0.9960
+  No improvement for 2/20 epochs
+Epoch 33: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [32:14<00:00,  1.60s/it, loss=0.00426] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:19<00:00,  1.39it/s] 
+
+Epoch 34/100
+  Train Loss: 0.0045
+  Val Loss: 0.0022
+  Val PCK: 0.9996
+  Val IoU: 0.9961
+  No improvement for 3/20 epochs
+Epoch 34: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [32:17<00:00,  1.60s/it, loss=0.00387] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:20<00:00,  1.38it/s] 
+
+Epoch 35/100
+  Train Loss: 0.0038
+  Val Loss: 0.0022
+  Val PCK: 0.9995
+  Val IoU: 0.9960
+  No improvement for 4/20 epochs
+Epoch 35: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [32:10<00:00,  1.60s/it, loss=0.00448] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:16<00:00,  1.40it/s] 
+
+Epoch 36/100
+  Train Loss: 0.0037
+  Val Loss: 0.0058
+  Val PCK: 0.9976
+  Val IoU: 0.9940
+  No improvement for 5/20 epochs
+Epoch 36: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [32:08<00:00,  1.59s/it, loss=0.00335] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:16<00:00,  1.40it/s] 
+
+Epoch 37/100
+  Train Loss: 0.0042
+  Val Loss: 0.0023
+  Val PCK: 0.9996
+  Val IoU: 0.9959
+  No improvement for 6/20 epochs
+Epoch 37: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [32:11<00:00,  1.60s/it, loss=0.00228] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:17<00:00,  1.39it/s] 
+
+Epoch 38/100
+  Train Loss: 0.0042
+  Val Loss: 0.0020
+  Val PCK: 0.9995
+  Val IoU: 0.9961
+  No improvement for 7/20 epochs
+Epoch 38: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [32:03<00:00,  1.59s/it, loss=0.00321] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:22<00:00,  1.38it/s] 
+
+Epoch 39/100
+  Train Loss: 0.0037
+  Val Loss: 0.0021
+  Val PCK: 0.9995
+  Val IoU: 0.9961
+  No improvement for 8/20 epochs
+Epoch 39: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [31:56<00:00,  1.58s/it, loss=0.00433] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:17<00:00,  1.39it/s] 
+
+Epoch 40/100
+  Train Loss: 0.0038
+  Val Loss: 0.0026
+  Val PCK: 0.9996
+  Val IoU: 0.9961
+  No improvement for 9/20 epochs
+Epoch 40: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [31:56<00:00,  1.58s/it, loss=0.00479]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:16<00:00,  1.40it/s] 
+
+Epoch 41/100
+  Train Loss: 0.0036
+  Val Loss: 0.0022
+  Val PCK: 0.9993
+  Val IoU: 0.9961
+  No improvement for 10/20 epochs
+Epoch 41: 100%|██████████████████████████████████████████████████████████████████████��███████████████| 1210/1210 [10:01<00:00,  2.01it/s, loss=0.00441] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.36it/s] 
+
+Epoch 42/100
+  Train Loss: 0.0037
+  Val Loss: 0.0022
+  Val PCK: 0.9996
+  Val IoU: 0.9962
+  No improvement for 11/20 epochs
+Epoch 42: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.02it/s, loss=0.00236] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:00<00:00,  4.37it/s] 
+
+Epoch 43/100
+  Train Loss: 0.0036
+  Val Loss: 0.0019
+  Val PCK: 0.9995
+  Val IoU: 0.9963
+  ✓ Saved best model (val_loss=0.0019)
+Epoch 43: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:00<00:00,  2.02it/s, loss=0.00302] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [03:08<00:00,  2.79it/s] 
+
+Epoch 44/100
+  Train Loss: 0.0036
+  Val Loss: 0.0026
+  Val PCK: 0.9993
+  Val IoU: 0.9961
+  No improvement for 1/20 epochs
+Epoch 44: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [32:04<00:00,  1.59s/it, loss=0.00309] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:19<00:00,  1.39it/s] 
+
+Epoch 45/100
+  Train Loss: 0.0035
+  Val Loss: 0.0020
+  Val PCK: 0.9994
+  Val IoU: 0.9963
+  No improvement for 2/20 epochs
+Epoch 45: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [31:41<00:00,  1.57s/it, loss=0.00345] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:15<00:00,  1.40it/s] 
+
+Epoch 46/100
+  Train Loss: 0.0035
+  Val Loss: 0.0019
+  Val PCK: 0.9996
+  Val IoU: 0.9962
+  No improvement for 3/20 epochs
+Epoch 46: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:57<00:00,  1.68s/it, loss=0.00432] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:04<00:00,  1.44it/s] 
+
+Epoch 47/100
+  Train Loss: 0.0036
+  Val Loss: 0.0024
+  Val PCK: 0.9996
+  Val IoU: 0.9961
+  No improvement for 4/20 epochs
+Epoch 47: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [17:26<00:00,  1.16it/s, loss=0.0044] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:08<00:00,  4.10it/s] 
+
+Epoch 48/100
+  Train Loss: 0.0034
+  Val Loss: 0.0017
+  Val PCK: 0.9996
+  Val IoU: 0.9964
+  ✓ Saved best model (val_loss=0.0017)
+Epoch 48: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:32<00:00,  1.91it/s, loss=0.00291]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:07<00:00,  4.11it/s] 
+
+Epoch 49/100
+  Train Loss: 0.0035
+  Val Loss: 0.0018
+  Val PCK: 0.9996
+  Val IoU: 0.9964
+  No improvement for 1/20 epochs
+Epoch 49: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:31<00:00,  1.92it/s, loss=0.00238] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:07<00:00,  4.11it/s] 
+
+Epoch 50/100
+  Train Loss: 0.0035
+  Val Loss: 0.0022
+  Val PCK: 0.9993
+  Val IoU: 0.9962
+  No improvement for 2/20 epochs
+Epoch 50: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:32<00:00,  1.91it/s, loss=0.00336]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:07<00:00,  4.12it/s] 
+
+Epoch 51/100
+  Train Loss: 0.0035
+  Val Loss: 0.0023
+  Val PCK: 0.9995
+  Val IoU: 0.9962
+  No improvement for 3/20 epochs
+Epoch 51: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:32<00:00,  1.91it/s, loss=0.0035] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:07<00:00,  4.12it/s] 
+
+Epoch 52/100
+  Train Loss: 0.0035
+  Val Loss: 0.0018
+  Val PCK: 0.9997
+  Val IoU: 0.9964
+  No improvement for 4/20 epochs
+Epoch 52: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:32<00:00,  1.91it/s, loss=0.00261] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [02:07<00:00,  4.12it/s] 
+
+Epoch 53/100
+  Train Loss: 0.0034
+  Val Loss: 0.0015
+  Val PCK: 0.9998
+  Val IoU: 0.9966
+  ✓ Saved best model (val_loss=0.0015)
+Epoch 53: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [10:31<00:00,  1.92it/s, loss=0.00357]
+Validation: 100%|█████████████████████████████████████████████████████████████████████████████████████████��██████████| 526/526 [02:08<00:00,  4.10it/s] 
+
+Epoch 54/100
+  Train Loss: 0.0034
+  Val Loss: 0.0015
+  Val PCK: 0.9998
+  Val IoU: 0.9966
+  No improvement for 1/20 epochs
+Epoch 54: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [19:13<00:00,  1.05it/s, loss=0.00175] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:10<00:00,  1.42it/s] 
+
+Epoch 55/100
+  Train Loss: 0.0033
+  Val Loss: 0.0016
+  Val PCK: 0.9995
+  Val IoU: 0.9965
+  No improvement for 2/20 epochs
+Epoch 55: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:46<00:00,  1.67s/it, loss=0.00286] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:03<00:00,  1.45it/s] 
+
+Epoch 56/100
+  Train Loss: 0.0032
+  Val Loss: 0.0017
+  Val PCK: 0.9996
+  Val IoU: 0.9964
+  No improvement for 3/20 epochs
+Epoch 56: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:44<00:00,  1.67s/it, loss=0.00308] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:06<00:00,  1.44it/s] 
+
+Epoch 57/100
+  Train Loss: 0.0033
+  Val Loss: 0.0018
+  Val PCK: 0.9996
+  Val IoU: 0.9965
+  No improvement for 4/20 epochs
+Epoch 57: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:40<00:00,  1.67s/it, loss=0.00391] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:10<00:00,  1.42it/s] 
+
+Epoch 58/100
+  Train Loss: 0.0032
+  Val Loss: 0.0016
+  Val PCK: 0.9996
+  Val IoU: 0.9965
+  No improvement for 5/20 epochs
+Epoch 58: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:43<00:00,  1.67s/it, loss=0.00334] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:07<00:00,  1.43it/s] 
+
+Epoch 59/100
+  Train Loss: 0.0032
+  Val Loss: 0.0013
+  Val PCK: 0.9999
+  Val IoU: 0.9966
+  ✓ Saved best model (val_loss=0.0013)
+Epoch 59: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:37<00:00,  1.67s/it, loss=0.00275]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:07<00:00,  1.43it/s] 
+
+Epoch 60/100
+  Train Loss: 0.0033
+  Val Loss: 0.0014
+  Val PCK: 0.9998
+  Val IoU: 0.9967
+  No improvement for 1/20 epochs
+Epoch 60: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:48<00:00,  1.68s/it, loss=0.00286]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:07<00:00,  1.43it/s] 
+
+Epoch 61/100
+  Train Loss: 0.0032
+  Val Loss: 0.0019
+  Val PCK: 0.9997
+  Val IoU: 0.9964
+  No improvement for 2/20 epochs
+Epoch 61: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:47<00:00,  1.68s/it, loss=0.00446] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:06<00:00,  1.43it/s] 
+
+Epoch 62/100
+  Train Loss: 0.0032
+  Val Loss: 0.0015
+  Val PCK: 0.9998
+  Val IoU: 0.9966
+  No improvement for 3/20 epochs
+Epoch 62: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:44<00:00,  1.67s/it, loss=0.00267] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:05<00:00,  1.44it/s] 
+
+Epoch 63/100
+  Train Loss: 0.0032
+  Val Loss: 0.0014
+  Val PCK: 0.9997
+  Val IoU: 0.9966
+  No improvement for 4/20 epochs
+Epoch 63: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:44<00:00,  1.67s/it, loss=0.00254] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:07<00:00,  1.43it/s] 
+
+Epoch 64/100
+  Train Loss: 0.0031
+  Val Loss: 0.0015
+  Val PCK: 0.9998
+  Val IoU: 0.9966
+  No improvement for 5/20 epochs
+Epoch 64: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:44<00:00,  1.67s/it, loss=0.00331] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:07<00:00,  1.43it/s] 
+
+Epoch 65/100
+  Train Loss: 0.0031
+  Val Loss: 0.0015
+  Val PCK: 0.9996
+  Val IoU: 0.9967
+  No improvement for 6/20 epochs
+Epoch 65: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:41<00:00,  1.67s/it, loss=0.00192] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:04<00:00,  1.44it/s] 
+
+Epoch 66/100
+  Train Loss: 0.0031
+  Val Loss: 0.0014
+  Val PCK: 0.9997
+  Val IoU: 0.9967
+  No improvement for 7/20 epochs
+Epoch 66: 100%|█████████████████████████████████████████████████████████████��████████████████████████| 1210/1210 [33:51<00:00,  1.68s/it, loss=0.00263] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:07<00:00,  1.43it/s] 
+
+Epoch 67/100
+  Train Loss: 0.0030
+  Val Loss: 0.0015
+  Val PCK: 0.9995
+  Val IoU: 0.9966
+  No improvement for 8/20 epochs
+Epoch 67: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:48<00:00,  1.68s/it, loss=0.00237] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:09<00:00,  1.42it/s] 
+
+Epoch 68/100
+  Train Loss: 0.0031
+  Val Loss: 0.0015
+  Val PCK: 0.9995
+  Val IoU: 0.9967
+  No improvement for 9/20 epochs
+Epoch 68: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:46<00:00,  1.68s/it, loss=0.00166] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:07<00:00,  1.43it/s] 
+
+Epoch 69/100
+  Train Loss: 0.0031
+  Val Loss: 0.0015
+  Val PCK: 0.9996
+  Val IoU: 0.9966
+  No improvement for 10/20 epochs
+Epoch 69: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:54<00:00,  1.68s/it, loss=0.00324] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:06<00:00,  1.43it/s] 
+
+Epoch 70/100
+  Train Loss: 0.0030
+  Val Loss: 0.0015
+  Val PCK: 0.9996
+  Val IoU: 0.9966
+  No improvement for 11/20 epochs
+Epoch 70: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:48<00:00,  1.68s/it, loss=0.00276]
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:07<00:00,  1.43it/s] 
+
+Epoch 71/100
+  Train Loss: 0.0030
+  Val Loss: 0.0015
+  Val PCK: 0.9996
+  Val IoU: 0.9967
+  No improvement for 12/20 epochs
+Epoch 71: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:54<00:00,  1.68s/it, loss=0.00347] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:03<00:00,  1.45it/s] 
+
+Epoch 72/100
+  Train Loss: 0.0030
+  Val Loss: 0.0015
+  Val PCK: 0.9998
+  Val IoU: 0.9967
+  No improvement for 13/20 epochs
+Epoch 72: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:52<00:00,  1.68s/it, loss=0.00197] 
+Validation: 100%|███████████████████��████████████████████████████████████████████████████████████████████████████████| 526/526 [06:07<00:00,  1.43it/s] 
+
+Epoch 73/100
+  Train Loss: 0.0030
+  Val Loss: 0.0014
+  Val PCK: 0.9996
+  Val IoU: 0.9967
+  No improvement for 14/20 epochs
+Epoch 73: 100%|███████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:54<00:00,  1.68s/it, loss=0.0027] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:06<00:00,  1.43it/s] 
+
+Epoch 74/100
+  Train Loss: 0.0030
+  Val Loss: 0.0014
+  Val PCK: 0.9997
+  Val IoU: 0.9967
+  No improvement for 15/20 epochs
+Epoch 74: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:48<00:00,  1.68s/it, loss=0.00273] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:07<00:00,  1.43it/s] 
+
+Epoch 75/100
+  Train Loss: 0.0030
+  Val Loss: 0.0014
+  Val PCK: 0.9996
+  Val IoU: 0.9966
+  No improvement for 16/20 epochs
+Epoch 75: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:46<00:00,  1.67s/it, loss=0.00345] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:05<00:00,  1.44it/s] 
+
+Epoch 76/100
+  Train Loss: 0.0029
+  Val Loss: 0.0017
+  Val PCK: 0.9996
+  Val IoU: 0.9966
+  No improvement for 17/20 epochs
+Epoch 76: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:43<00:00,  1.67s/it, loss=0.00317] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:05<00:00,  1.44it/s] 
+
+Epoch 77/100
+  Train Loss: 0.0029
+  Val Loss: 0.0014
+  Val PCK: 0.9997
+  Val IoU: 0.9967
+  No improvement for 18/20 epochs
+Epoch 77: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:44<00:00,  1.67s/it, loss=0.00289] 
+Validation: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 526/526 [06:07<00:00,  1.43it/s] 
+
+Epoch 78/100
+  Train Loss: 0.0030
+  Val Loss: 0.0014
+  Val PCK: 0.9996
+  Val IoU: 0.9967
+  No improvement for 19/20 epochs
+Epoch 78: 100%|██████████████████████████████████████████████████████████████████████████████████████| 1210/1210 [33:53<00:00,  1.68s/it, loss=0.00345] 
+Validation: 100%|█████████████████████████████████████████████████████████████████████████████████████��██████████████| 526/526 [06:04<00:00,  1.44it/s] 
+
+Epoch 79/100
+  Train Loss: 0.0029
+  Val Loss: 0.0013
+  Val PCK: 0.9997
+  Val IoU: 0.9967
+  No improvement for 20/20 epochs
+
+⚡ Early stopping triggered after 79 epochs!
+wandb: 
+wandb: Run history:
+wandb:               batch ▃▂▃▅▇▇▅█▇▇█▃▇▄▃▁▃▃▅▇▅▂▄▆▂▇▂▂▂▃▅▆▆█▃▅▆▇▅▇
+wandb:               epoch ▁▁▁▁▁▂▂▂▂▂▂▃▃▃▃▄▄▄▄▄▅▅▅▅▅▅▆▆▆▆▆▆▆▆▆▇▇▇▇█
+wandb: finetune/epoch_loss █▂▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁
+wandb:    finetune/l_angle ▁▄▇▁▁▁▁▁█▁▁▂▁▂▂▇▁▁▁▁▁▁▁▁▂▁▁▁▁▁▁▁▂▁▁▁▁▁▁▂
+wandb:     finetune/l_diag ▁█▂▁▂▁▁▁▂▁▁▁▁▁▁▅▁▁▁▁█▁▁▁▁▁▁▁▁▁▂▁▁▁▁▁▁▂▂▁
+wandb:     finetune/l_edge ▆▁▁▁▂▁▁▁▄▄▁▁▁▁▁▁▁▅▁▁█▁▁▁▂▄▂▁▂▁▆▁▁▁▁▃▁▁▁▁
+wandb:       finetune/l_hm █▆▄▄▄▅▅▃▄▄▃▃▃▃▃▂▃▄▂▂▂▂▂▁▁▂▂▁▁▂▁▁▂▁▁▁▁▁▁▁
+wandb:      finetune/l_kpt ▆▃▁▁█▁▁▁▁▁▂▂▁▁▁▁▁▂▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▂▁▁▁▁▁▁
+wandb:       finetune/loss █▄▃▂▄▂▂▂▄▄▁▁▁▃▄▄▅▁▃▁▁▂▁▂▁▃▁▄▁▁▁▁▁▁▁▁▄▁▂▁
+wandb:         finetune/lr ▃▅████████▇▇▇▇▇▆▆▅▅▅▄▄▄▄▄▄▄▄▄▄▂▂▂▂▂▁▁▁▁▁
+wandb:                  +5 ...
+wandb:
+wandb: Run summary:
+wandb:               batch 1200
+wandb:          best_epoch 59
+wandb:       best_val_loss 0.0013
+wandb:               epoch 79
+wandb: finetune/epoch_loss 0.00295
+wandb:    finetune/l_angle 2e-05
+wandb:     finetune/l_diag 0.0
+wandb:     finetune/l_edge 1e-05
+wandb:       finetune/l_hm 3e-05
+wandb:      finetune/l_kpt 2e-05
+wandb:                  +7 ...
+wandb:
+wandb:  View run Finetune_01/29_06:15 at: https://wandb.ai/adithyanraj03/courtkeynet/runs/0fnvysj4
+wandb:  View project at: https://wandb.ai/adithyanraj03/courtkeynet
+wandb: Synced 5 W&B file(s), 80 media file(s), 2 artifact file(s) and 0 other file(s)
+wandb: Find logs at: .\wandb\run-20260129_061557-0fnvysj4\logs
+
+============================================================
+  Fine-tuning Complete!
+  Best val loss: 0.0013
+  Weights saved to: runs\courtkeynet_finetune\finetune_20260129_061601_0fnvysj4
+============================================================

pretrained/courtkeynet_base.safetensors

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

pretrained/training_log.txt

@@ -0,0 +1,1026 @@
+  Train Loss: 0.0851
+  Val Loss: 0.1043
+  Val PCK: 0.9900
+  Val IoU: 0.9398
+  No improvement for 1/20 epochs
+Epoch 48: 100%|██████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [19:36<00:00,  2.00it/s, loss=0.0841] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:28<00:00,  5.25it/s] 
+
+Epoch 49/300
+  Train Loss: 0.0846
+  Val Loss: 0.1037
+  Val PCK: 0.9902
+  Val IoU: 0.9401
+  No improvement for 2/20 epochs
+Epoch 49: 100%|██████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [19:23<00:00,  2.03it/s, loss=0.0846]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:28<00:00,  5.29it/s] 
+
+Epoch 50/300
+  Train Loss: 0.0845
+  Val Loss: 0.1036
+  Val PCK: 0.9901
+  Val IoU: 0.9408
+  Saved best model (val_loss=0.1036)
+Epoch 50: 100%|███████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [19:31<00:00,  2.01it/s, loss=0.086]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:31<00:00,  5.07it/s] 
+
+Epoch 51/300
+  Train Loss: 0.0849
+  Val Loss: 0.1043
+  Val PCK: 0.9901
+  Val IoU: 0.9394
+  No improvement for 1/20 epochs
+Epoch 51: 100%|██████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [19:17<00:00,  2.04it/s, loss=0.0833] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:28<00:00,  5.26it/s] 
+
+Epoch 52/300
+  Train Loss: 0.0840
+  Val Loss: 0.1040
+  Val PCK: 0.9905
+  Val IoU: 0.9404
+  No improvement for 2/20 epochs
+Epoch 52: 100%|███████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [19:30<00:00,  2.01it/s, loss=0.083] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:27<00:00,  5.32it/s] 
+
+Epoch 53/300
+  Train Loss: 0.0838
+  Val Loss: 0.1037
+  Val PCK: 0.9896
+  Val Loss: 0.1045
+  Val PCK: 0.9897
+  Val IoU: 0.9396
+  No improvement for 4/20 epochs
+Epoch 54: 100%|██████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [19:26<00:00,  2.02it/s, loss=0.0849]     
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:30<00:00,  5.16it/s]     
+
+Epoch 55/300
+  Train Loss: 0.0829
+  Val Loss: 0.1067
+  Val PCK: 0.9901
+  Val IoU: 0.9382
+  No improvement for 5/20 epochs
+Epoch 55: 100%|██████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [19:28<00:00,  2.02it/s, loss=0.0827]     
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:27<00:00,  5.34it/s]     
+
+Epoch 56/300
+  Train Loss: 0.0826
+  Val Loss: 0.1044
+  Val PCK: 0.9893
+  Val IoU: 0.9396
+  No improvement for 6/20 epochs
+Epoch 56: 100%|██████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [19:17<00:00,  2.04it/s, loss=0.0837]     
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:18<00:00,  5.90it/s]     
+
+Epoch 57/300
+  Train Loss: 0.0825
+  Val Loss: 0.1036
+  Val PCK: 0.9909
+  Val IoU: 0.9401
+  No improvement for 7/20 epochs
+Epoch 57:  13%|███████████                                                                            | 300/2357 [02:36<17:02,  2.01it/s, loss=0.08Epoch 57:  13%|███████████                                                                            | 300/2357 [02:37<17:02,  2.01it/s, loss=0.08Epoch 57:  13%EpEpoch 57: 100%|██████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [19:20<00:00,  2.03it/s, loss=0.0822]     
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:29<00:00,  5.20it/s]
+
+Epoch 58/300
+  Train Loss: 0.0825
+  Val Loss: 0.1024
+  Val PCK: 0.9901
+  Val IoU: 0.9409
+  Saved best model (val_loss=0.1024)
+Epoch 58: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [19:33<00:00,  2.01it/s, loss=0.0814]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:31<00:00,  5.11it/s] 
+
+Epoch 59/300
+  Train Loss: 0.0819
+  Val Loss: 0.1019
+  Val PCK: 0.9909
+  Val IoU: 0.9416
+  Saved best model (val_loss=0.1019)
+Epoch 59: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:35<00:00,  2.11it/s, loss=0.0809]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:22<00:00,  5.62it/s] 
+
+Epoch 60/300
+  Train Loss: 0.0819
+  Val Loss: 0.1028
+  Val PCK: 0.9907
+  Val IoU: 0.9403
+  No improvement for 1/20 epochs
+Epoch 60: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:50<00:00,  2.08it/s, loss=0.0836]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.85it/s] 
+
+Epoch 61/300
+  Train Loss: 0.0815
+  Val Loss: 0.1050
+  Val PCK: 0.9889
+  Val IoU: 0.9397
+  No improvement for 2/20 epochs
+Epoch 61: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:54<00:00,  2.08it/s, loss=0.0805] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.86it/s] 
+
+Epoch 62/300
+  Train Loss: 0.0813
+  Val Loss: 0.1039
+  Val PCK: 0.9883
+  Val IoU: 0.9402
+  No improvement for 3/20 epochs
+Epoch 62: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:15<00:00,  2.15it/s, loss=0.0814] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:04<00:00,  7.24it/s] 
+
+Epoch 63/300
+  Train Loss: 0.0806
+  Val Loss: 0.1029
+  Val PCK: 0.9901
+  Val IoU: 0.9409
+  No improvement for 4/20 epochs
+Epoch 63: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:37<00:00,  2.69it/s, loss=0.0826] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:05<00:00,  7.17it/s] 
+
+Epoch 64/300
+  Train Loss: 0.0809
+  Val Loss: 0.1032
+  Val PCK: 0.9888
+  Val IoU: 0.9407
+  No improvement for 5/20 epochs
+Epoch 64: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:36<00:00,  2.69it/s, loss=0.0811] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:04<00:00,  7.27it/s] 
+
+Epoch 65/300
+  Train Loss: 0.0803
+  Val Loss: 0.1012
+  Val PCK: 0.9907
+  Val IoU: 0.9411
+  Saved best model (val_loss=0.1012)
+Epoch 65: 100%|███████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:35<00:00,  2.69it/s, loss=0.082]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:04<00:00,  7.25it/s] 
+
+Epoch 66/300
+  Train Loss: 0.0801
+  Val Loss: 0.1036
+  Val PCK: 0.9888
+  Val IoU: 0.9404
+  No improvement for 1/20 epochs
+Epoch 66: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:35<00:00,  2.69it/s, loss=0.0805] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:04<00:00,  7.22it/s] 
+
+Epoch 67/300
+  Train Loss: 0.0799
+  Val Loss: 0.1013
+  Val PCK: 0.9907
+  Val IoU: 0.9416
+  No improvement for 2/20 epochs
+Epoch 67: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:36<00:00,  2.69it/s, loss=0.0801] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:04<00:00,  7.22it/s] 
+
+Epoch 68/300
+  Train Loss: 0.0795
+  Val Loss: 0.1016
+  Val PCK: 0.9901
+  Val IoU: 0.9412
+  No improvement for 3/20 epochs
+Epoch 68: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:36<00:00,  2.69it/s, loss=0.0796] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:04<00:00,  7.24it/s] 
+
+Epoch 69/300
+  Train Loss: 0.0792
+  Val Loss: 0.1022
+  Val PCK: 0.9901
+  Val IoU: 0.9413
+  No improvement for 4/20 epochs
+Epoch 69: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [17:34<00:00,  2.23it/s, loss=0.0802] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.87it/s] 
+
+Epoch 70/300
+  Train Loss: 0.0791
+  Val Loss: 0.1022
+  Val PCK: 0.9914
+  Val IoU: 0.9411
+  No improvement for 5/20 epochs
+Epoch 70: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:46<00:00,  2.09it/s, loss=0.0795]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.87it/s] 
+
+Epoch 71/300
+  Train Loss: 0.0791
+  Val Loss: 0.1043
+  Val PCK: 0.9896
+  Val IoU: 0.9403
+  No improvement for 6/20 epochs
+Epoch 71: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:43<00:00,  2.10it/s, loss=0.0793] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.86it/s] 
+
+Epoch 72/300
+  Train Loss: 0.0791
+  Val Loss: 0.1025
+  Val PCK: 0.9895
+  Val IoU: 0.9410
+  No improvement for 7/20 epochs
+Epoch 72: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:44<00:00,  2.10it/s, loss=0.0791] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.88it/s] 
+
+Epoch 73/300
+  Train Loss: 0.0785
+  Val Loss: 0.1025
+  Val PCK: 0.9898
+  Val IoU: 0.9405
+  No improvement for 8/20 epochs
+Epoch 73: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:46<00:00,  2.09it/s, loss=0.0834] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.88it/s] 
+
+Epoch 74/300
+  Train Loss: 0.0788
+  Val Loss: 0.1035
+  Val PCK: 0.9894
+  Val IoU: 0.9403
+  No improvement for 9/20 epochs
+Epoch 74: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:44<00:00,  2.10it/s, loss=0.0771] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:18<00:00,  5.90it/s] 
+
+Epoch 75/300
+  Train Loss: 0.0784
+  Val Loss: 0.1013
+  Val PCK: 0.9898
+  Val IoU: 0.9416
+  No improvement for 10/20 epochs
+Epoch 75: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:46<00:00,  2.09it/s, loss=0.0777]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.89it/s] 
+
+Epoch 76/300
+  Train Loss: 0.0784
+  Val Loss: 0.1025
+  Val PCK: 0.9906
+  Val IoU: 0.9410
+  No improvement for 11/20 epochs
+Epoch 76: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:45<00:00,  2.09it/s, loss=0.0778] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:18<00:00,  5.90it/s] 
+
+Epoch 77/300
+  Train Loss: 0.0778
+  Val Loss: 0.1008
+  Val PCK: 0.9914
+  Val IoU: 0.9418
+  Saved best model (val_loss=0.1008)
+Epoch 77: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:45<00:00,  2.09it/s, loss=0.0765]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.90it/s] 
+
+Epoch 78/300
+  Train Loss: 0.0772
+  Val Loss: 0.1011
+  Val PCK: 0.9914
+  Val IoU: 0.9415
+  No improvement for 1/20 epochs
+Epoch 78: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:45<00:00,  2.09it/s, loss=0.0772] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.88it/s] 
+
+Epoch 79/300
+  Train Loss: 0.0777
+  Val Loss: 0.1007
+  Val PCK: 0.9910
+  Val IoU: 0.9417
+  Saved best model (val_loss=0.1007)
+Epoch 79: 100%|███████████��██████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:44<00:00,  2.10it/s, loss=0.0768]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.89it/s] 
+
+Epoch 80/300
+  Train Loss: 0.0776
+  Val Loss: 0.1001
+  Val PCK: 0.9916
+  Val IoU: 0.9419
+  Saved best model (val_loss=0.1001)
+Epoch 80: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:44<00:00,  2.10it/s, loss=0.0783]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.89it/s] 
+
+Epoch 81/300
+  Train Loss: 0.0778
+  Val Loss: 0.0998
+  Val PCK: 0.9912
+  Val IoU: 0.9426
+  Saved best model (val_loss=0.0998)
+Epoch 81: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:44<00:00,  2.10it/s, loss=0.0773]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.88it/s] 
+
+Epoch 82/300
+  Train Loss: 0.0766
+  Val Loss: 0.1029
+  Val PCK: 0.9907
+  Val IoU: 0.9398
+  No improvement for 1/20 epochs
+Epoch 82: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:44<00:00,  2.10it/s, loss=0.0763] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.90it/s] 
+
+Epoch 83/300
+  Train Loss: 0.0767
+  Val Loss: 0.1020
+  Val PCK: 0.9900
+  Val IoU: 0.9412
+  No improvement for 2/20 epochs
+Epoch 83: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:43<00:00,  2.10it/s, loss=0.0799] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:19<00:00,  5.88it/s] 
+
+Epoch 84/300
+  Train Loss: 0.0770
+  Val Loss: 0.1022
+  Val PCK: 0.9905
+  Val IoU: 0.9409
+  No improvement for 3/20 epochs
+Epoch 84: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [18:43<00:00,  2.10it/s, loss=0.0766] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:18<00:00,  5.90it/s] 
+
+Epoch 85/300
+  Train Loss: 0.0766
+  Val Loss: 0.1006
+  Val PCK: 0.9917
+  Val IoU: 0.9418
+  No improvement for 4/20 epochs
+Epoch 85: 100%|██████████████████████████████████████��███████████████████████████████████████████████████| 2357/2357 [19:51<00:00,  1.98it/s, loss=0.0764]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.46it/s] 
+
+Epoch 86/300
+  Train Loss: 0.0766
+  Val Loss: 0.1012
+  Val PCK: 0.9906
+  Val IoU: 0.9419
+  No improvement for 5/20 epochs
+Epoch 86: 100%|███████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.65it/s, loss=0.076] 
+Validation:   0%|                                                                                                                 | 0/466 [00:00<?, ?it/s]C:\Users\rajur\miniconda3\envs\gpu\Lib\site-packages\albumentations\check_version.py:147: UserWarning: Error fetching version info The read operation timed out
+  data = fetch_version_info()
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:08<00:00,  6.83it/s]
+
+Epoch 87/300
+  Train Loss: 0.0764
+  Val Loss: 0.1021
+  Val PCK: 0.9904
+  Val IoU: 0.9415
+  No improvement for 6/20 epochs
+Epoch 87: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0807] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.46it/s] 
+
+Epoch 88/300
+  Train Loss: 0.0761
+  Val Loss: 0.1008
+  Val PCK: 0.9911
+  Val IoU: 0.9421
+  No improvement for 7/20 epochs
+Epoch 88: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0754] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.47it/s] 
+
+Epoch 89/300
+  Train Loss: 0.0760
+  Val Loss: 0.1008
+  Val PCK: 0.9904
+  Val IoU: 0.9416
+  No improvement for 8/20 epochs
+Epoch 89: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:52<00:00,  2.64it/s, loss=0.0742] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 90/300
+  Train Loss: 0.0757
+  Val Loss: 0.1011
+  Val PCK: 0.9888
+  Val IoU: 0.9419
+  No improvement for 9/20 epochs
+Epoch 90: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0749]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 91/300
+  Train Loss: 0.0755
+  Val Loss: 0.1005
+  Val PCK: 0.9899
+  Val IoU: 0.9420
+  No improvement for 10/20 epochs
+Epoch 91: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0765] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 92/300
+  Train Loss: 0.0757
+  Val Loss: 0.1001
+  Val PCK: 0.9904
+  Val IoU: 0.9420
+  No improvement for 11/20 epochs
+Epoch 92: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0755] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.40it/s] 
+
+Epoch 93/300
+  Train Loss: 0.0752
+  Val Loss: 0.1013
+  Val PCK: 0.9897
+  Val IoU: 0.9416
+  No improvement for 12/20 epochs
+Epoch 93: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0754] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.43it/s] 
+
+Epoch 94/300
+  Train Loss: 0.0754
+  Val Loss: 0.1005
+  Val PCK: 0.9904
+  Val IoU: 0.9414
+  No improvement for 13/20 epochs
+Epoch 94: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0755] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.44it/s] 
+
+Epoch 95/300
+  Train Loss: 0.0754
+  Val Loss: 0.1008
+  Val PCK: 0.9907
+  Val IoU: 0.9418
+  No improvement for 14/20 epochs
+Epoch 95: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0731]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:03<00:00,  7.39it/s] 
+
+Epoch 96/300
+  Train Loss: 0.0749
+  Val Loss: 0.1001
+  Val PCK: 0.9915
+  Val IoU: 0.9417
+  No improvement for 15/20 epochs
+Epoch 96: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0767] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.40it/s] 
+
+Epoch 97/300
+  Train Loss: 0.0747
+  Val Loss: 0.1010
+  Val PCK: 0.9899
+  Val IoU: 0.9420
+  No improvement for 16/20 epochs
+Epoch 97: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0747] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 98/300
+  Train Loss: 0.0744
+  Val Loss: 0.1005
+  Val PCK: 0.9905
+  Val IoU: 0.9418
+  No improvement for 17/20 epochs
+Epoch 98: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0737] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 99/300
+  Train Loss: 0.0744
+  Val Loss: 0.0991
+  Val PCK: 0.9914
+  Val IoU: 0.9428
+  Saved best model (val_loss=0.0991)
+Epoch 99: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0734] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 100/300
+  Train Loss: 0.0746
+  Val Loss: 0.0994
+  Val PCK: 0.9916
+  Val IoU: 0.9421
+  No improvement for 1/20 epochs
+Epoch 100: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0759]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 101/300
+  Train Loss: 0.0746
+  Val Loss: 0.1013
+  Val PCK: 0.9898
+  Val IoU: 0.9415
+  No improvement for 2/20 epochs
+Epoch 101: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0741] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.43it/s] 
+
+Epoch 102/300
+  Train Loss: 0.0743
+  Val Loss: 0.0985
+  Val PCK: 0.9914
+  Val IoU: 0.9429
+  Saved best model (val_loss=0.0985)
+Epoch 102: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0743] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 103/300
+  Train Loss: 0.0741
+  Val Loss: 0.0995
+  Val PCK: 0.9910
+  Val IoU: 0.9423
+  No improvement for 1/20 epochs
+Epoch 103: 100%|████��████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0753] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.40it/s] 
+
+Epoch 104/300
+  Train Loss: 0.0740
+  Val Loss: 0.0994
+  Val PCK: 0.9913
+  Val IoU: 0.9426
+  No improvement for 2/20 epochs
+Epoch 104: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0748] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:03<00:00,  7.38it/s] 
+
+Epoch 105/300
+  Train Loss: 0.0738
+  Val Loss: 0.1003
+  Val PCK: 0.9918
+  Val IoU: 0.9418
+  No improvement for 3/20 epochs
+Epoch 105: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0725]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 106/300
+  Train Loss: 0.0737
+  Val Loss: 0.0994
+  Val PCK: 0.9909
+  Val IoU: 0.9426
+  No improvement for 4/20 epochs
+Epoch 106: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0737] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:03<00:00,  7.39it/s] 
+
+Epoch 107/300
+  Train Loss: 0.0735
+  Val Loss: 0.0991
+  Val PCK: 0.9918
+  Val IoU: 0.9425
+  No improvement for 5/20 epochs
+Epoch 107: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0746] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 108/300
+  Train Loss: 0.0734
+  Val Loss: 0.0998
+  Val PCK: 0.9908
+  Val IoU: 0.9421
+  No improvement for 6/20 epochs
+Epoch 108: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0731] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 109/300
+  Train Loss: 0.0733
+  Val Loss: 0.0999
+  Val PCK: 0.9914
+  Val IoU: 0.9421
+  No improvement for 7/20 epochs
+Epoch 109: 100%|███████████████████████████████████��██████████████████████████████████████████████████████| 2357/2357 [14:52<00:00,  2.64it/s, loss=0.073] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:03<00:00,  7.39it/s] 
+
+Epoch 110/300
+  Train Loss: 0.0733
+  Val Loss: 0.0993
+  Val PCK: 0.9905
+  Val IoU: 0.9425
+  No improvement for 8/20 epochs
+Epoch 110: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0715]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.43it/s] 
+
+Epoch 111/300
+  Train Loss: 0.0731
+  Val Loss: 0.0994
+  Val PCK: 0.9905
+  Val IoU: 0.9423
+  No improvement for 9/20 epochs
+Epoch 111: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0724] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 112/300
+  Train Loss: 0.0730
+  Val Loss: 0.0989
+  Val PCK: 0.9911
+  Val IoU: 0.9427
+  No improvement for 10/20 epochs
+Epoch 112: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0738] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:03<00:00,  7.38it/s] 
+
+Epoch 113/300
+  Train Loss: 0.0733
+  Val Loss: 0.1001
+  Val PCK: 0.9907
+  Val IoU: 0.9416
+  No improvement for 11/20 epochs
+Epoch 113: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.073] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.44it/s] 
+
+Epoch 114/300
+  Train Loss: 0.0726
+  Val Loss: 0.0991
+  Val PCK: 0.9915
+  Val IoU: 0.9425
+  No improvement for 12/20 epochs
+Epoch 114: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0733] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.44it/s] 
+
+Epoch 115/300
+  Train Loss: 0.0728
+  Val Loss: 0.0989
+  Val PCK: 0.9907
+  Val IoU: 0.9430
+  No improvement for 13/20 epochs
+Epoch 115: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0741]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.44it/s] 
+
+Epoch 116/300
+  Train Loss: 0.0726
+  Val Loss: 0.0991
+  Val PCK: 0.9915
+  Val IoU: 0.9425
+  No improvement for 14/20 epochs
+Epoch 116: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0735] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.43it/s] 
+
+Epoch 117/300
+  Train Loss: 0.0724
+  Val Loss: 0.1004
+  Val PCK: 0.9909
+  Val IoU: 0.9416
+  No improvement for 15/20 epochs
+Epoch 117: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:52<00:00,  2.64it/s, loss=0.0732] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.46it/s] 
+
+Epoch 118/300
+  Train Loss: 0.0723
+  Val Loss: 0.0978
+  Val PCK: 0.9920
+  Val IoU: 0.9430
+  Saved best model (val_loss=0.0978)
+Epoch 118: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0722]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.44it/s] 
+
+Epoch 119/300
+  Train Loss: 0.0721
+  Val Loss: 0.0998
+  Val PCK: 0.9910
+  Val IoU: 0.9422
+  No improvement for 1/20 epochs
+Epoch 119: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:52<00:00,  2.64it/s, loss=0.0725] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.46it/s] 
+
+Epoch 120/300
+  Train Loss: 0.0720
+  Val Loss: 0.0994
+  Val PCK: 0.9904
+  Val IoU: 0.9420
+  No improvement for 2/20 epochs
+Epoch 120: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0724]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.47it/s] 
+
+Epoch 121/300
+  Train Loss: 0.0722
+  Val Loss: 0.1002
+  Val PCK: 0.9916
+  Val IoU: 0.9422
+  No improvement for 3/20 epochs
+Epoch 121: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0701] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.45it/s] 
+
+Epoch 122/300
+  Train Loss: 0.0717
+  Val Loss: 0.0989
+  Val PCK: 0.9906
+  Val IoU: 0.9424
+  No improvement for 4/20 epochs
+Epoch 122: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0696] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.46it/s] 
+
+Epoch 123/300
+  Train Loss: 0.0720
+  Val Loss: 0.0997
+  Val PCK: 0.9911
+  Val IoU: 0.9424
+  No improvement for 5/20 epochs
+Epoch 123: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0722] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.45it/s] 
+
+Epoch 124/300
+  Train Loss: 0.0719
+  Val Loss: 0.0995
+  Val PCK: 0.9910
+  Val IoU: 0.9423
+  No improvement for 6/20 epochs
+Epoch 124: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0716] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.43it/s] 
+
+Epoch 125/300
+  Train Loss: 0.0716
+  Val Loss: 0.0988
+  Val PCK: 0.9916
+  Val IoU: 0.9427
+  No improvement for 7/20 epochs
+Epoch 125: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:52<00:00,  2.64it/s, loss=0.071]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.43it/s] 
+
+Epoch 126/300
+  Train Loss: 0.0717
+  Val Loss: 0.0996
+  Val PCK: 0.9909
+  Val IoU: 0.9422
+  No improvement for 8/20 epochs
+Epoch 126: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:52<00:00,  2.64it/s, loss=0.0703] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 127/300
+  Train Loss: 0.0712
+  Val Loss: 0.0988
+  Val PCK: 0.9906
+  Val IoU: 0.9425
+  No improvement for 9/20 epochs
+Epoch 127: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0729] 
+Validation: 100%|███████████��███████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.43it/s] 
+
+Epoch 128/300
+  Train Loss: 0.0714
+  Val Loss: 0.0983
+  Val PCK: 0.9913
+  Val IoU: 0.9430
+  No improvement for 10/20 epochs
+Epoch 128: 100%|███████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.07] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.44it/s] 
+
+Epoch 129/300
+  Train Loss: 0.0712
+  Val Loss: 0.0993
+  Val PCK: 0.9904
+  Val IoU: 0.9423
+  No improvement for 11/20 epochs
+Epoch 129: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0709] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 130/300
+  Train Loss: 0.0712
+  Val Loss: 0.0980
+  Val PCK: 0.9917
+  Val IoU: 0.9431
+  No improvement for 12/20 epochs
+Epoch 130: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0716]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.47it/s] 
+
+Epoch 131/300
+  Train Loss: 0.0711
+  Val Loss: 0.0992
+  Val PCK: 0.9909
+  Val IoU: 0.9426
+  No improvement for 13/20 epochs
+Epoch 131: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0716] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.47it/s] 
+
+Epoch 132/300
+  Train Loss: 0.0710
+  Val Loss: 0.0997
+  Val PCK: 0.9905
+  Val IoU: 0.9422
+  No improvement for 14/20 epochs
+Epoch 132: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0713] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.43it/s] 
+
+Epoch 133/300
+  Train Loss: 0.0710
+  Val Loss: 0.0994
+  Val PCK: 0.9908
+  Val IoU: 0.9424
+  No improvement for 15/20 epochs
+Epoch 133: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.069] 
+Validation: 100%|██████████████████████████████████████��████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 134/300
+  Train Loss: 0.0709
+  Val Loss: 0.0993
+  Val PCK: 0.9913
+  Val IoU: 0.9423
+  No improvement for 16/20 epochs
+Epoch 134: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0716] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.43it/s] 
+
+Epoch 135/300
+  Train Loss: 0.0709
+  Val Loss: 0.0992
+  Val PCK: 0.9910
+  Val IoU: 0.9422
+  No improvement for 17/20 epochs
+Epoch 135: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0716]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.44it/s] 
+
+Epoch 136/300
+  Train Loss: 0.0708
+  Val Loss: 0.0986
+  Val PCK: 0.9900
+  Val IoU: 0.9427
+  No improvement for 18/20 epochs
+Epoch 136: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0703] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.48it/s] 
+
+Epoch 137/300
+  Train Loss: 0.0709
+  Val Loss: 0.0999
+  Val PCK: 0.9900
+  Val IoU: 0.9417
+  No improvement for 19/20 epochs
+Epoch 137: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.071] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 138/300
+  Train Loss: 0.0704
+  Val Loss: 0.0977
+  Val PCK: 0.9909
+  Val IoU: 0.9433
+  Saved best model (val_loss=0.0977)
+Epoch 138: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:55<00:00,  2.63it/s, loss=0.0702]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 139/300
+  Train Loss: 0.0705
+  Val Loss: 0.0989
+  Val PCK: 0.9908
+  Val IoU: 0.9426
+  No improvement for 1/20 epochs
+Epoch 139: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:52<00:00,  2.64it/s, loss=0.0699] 
+Validation: 100%|██████████████████████████████████████████████████████████████████��████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 140/300
+  Train Loss: 0.0704
+  Val Loss: 0.0990
+  Val PCK: 0.9906
+  Val IoU: 0.9425
+  No improvement for 2/20 epochs
+Epoch 140: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0694]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:03<00:00,  7.38it/s] 
+
+Epoch 141/300
+  Train Loss: 0.0704
+  Val Loss: 0.0977
+  Val PCK: 0.9916
+  Val IoU: 0.9430
+  No improvement for 3/20 epochs
+Epoch 141: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.069] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 142/300
+  Train Loss: 0.0704
+  Val Loss: 0.1005
+  Val PCK: 0.9905
+  Val IoU: 0.9416
+  No improvement for 4/20 epochs
+Epoch 142: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0688] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:03<00:00,  7.39it/s] 
+
+Epoch 143/300
+  Train Loss: 0.0701
+  Val Loss: 0.0989
+  Val PCK: 0.9909
+  Val IoU: 0.9428
+  No improvement for 5/20 epochs
+Epoch 143: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0678] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 144/300
+  Train Loss: 0.0700
+  Val Loss: 0.0998
+  Val PCK: 0.9901
+  Val IoU: 0.9425
+  No improvement for 6/20 epochs
+Epoch 144: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0724] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:03<00:00,  7.38it/s] 
+
+Epoch 145/300
+  Train Loss: 0.0703
+  Val Loss: 0.0992
+  Val PCK: 0.9910
+  Val IoU: 0.9425
+  No improvement for 7/20 epochs
+Epoch 145: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.65it/s, loss=0.0696]
+Validation: 100%|█████████████████████████████████████████████████████████████████████████████████████████████████���█████| 466/466 [01:02<00:00,  7.40it/s] 
+
+Epoch 146/300
+  Train Loss: 0.0699
+  Val Loss: 0.0993
+  Val PCK: 0.9919
+  Val IoU: 0.9422
+  No improvement for 8/20 epochs
+Epoch 146: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0695] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 147/300
+  Train Loss: 0.0697
+  Val Loss: 0.0981
+  Val PCK: 0.9926
+  Val IoU: 0.9427
+  No improvement for 9/20 epochs
+Epoch 147: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0677] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.45it/s] 
+
+Epoch 148/300
+  Train Loss: 0.0697
+  Val Loss: 0.1006
+  Val PCK: 0.9901
+  Val IoU: 0.9420
+  No improvement for 10/20 epochs
+Epoch 148: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0678] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:03<00:00,  7.39it/s] 
+
+Epoch 149/300
+  Train Loss: 0.0696
+  Val Loss: 0.0984
+  Val PCK: 0.9907
+  Val IoU: 0.9431
+  No improvement for 11/20 epochs
+Epoch 149: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0697] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 150/300
+  Train Loss: 0.0698
+  Val Loss: 0.0984
+  Val PCK: 0.9916
+  Val IoU: 0.9427
+  No improvement for 12/20 epochs
+Epoch 150: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0694]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.45it/s] 
+
+Epoch 151/300
+  Train Loss: 0.0696
+  Val Loss: 0.0982
+  Val PCK: 0.9915
+  Val IoU: 0.9430
+  No improvement for 13/20 epochs
+Epoch 151: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0716] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 152/300
+  Train Loss: 0.0696
+  Val Loss: 0.0989
+  Val PCK: 0.9910
+  Val IoU: 0.9426
+  No improvement for 14/20 epochs
+Epoch 152: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0682] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 153/300
+  Train Loss: 0.0691
+  Val Loss: 0.0979
+  Val PCK: 0.9912
+  Val IoU: 0.9431
+  No improvement for 15/20 epochs
+Epoch 153: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.068] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.42it/s] 
+
+Epoch 154/300
+  Train Loss: 0.0693
+  Val Loss: 0.1005
+  Val PCK: 0.9897
+  Val IoU: 0.9415
+  No improvement for 16/20 epochs
+Epoch 154: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [15:00<00:00,  2.62it/s, loss=0.0708] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 155/300
+  Train Loss: 0.0694
+  Val Loss: 0.0981
+  Val PCK: 0.9912
+  Val IoU: 0.9430
+  No improvement for 17/20 epochs
+Epoch 155: 100%|██████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.068]
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.41it/s] 
+
+Epoch 156/300
+  Train Loss: 0.0692
+  Val Loss: 0.0993
+  Val PCK: 0.9904
+  Val IoU: 0.9420
+  No improvement for 18/20 epochs
+Epoch 156: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0708] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:02<00:00,  7.43it/s] 
+
+Epoch 157/300
+  Train Loss: 0.0692
+  Val Loss: 0.0985
+  Val PCK: 0.9906
+  Val IoU: 0.9425
+  No improvement for 19/20 epochs
+Epoch 157: 100%|█████████████████████████████████████████████████████████████████████████████████████████| 2357/2357 [14:51<00:00,  2.64it/s, loss=0.0712] 
+Validation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████| 466/466 [01:03<00:00,  7.37it/s] 
+
+Epoch 158/300
+  Train Loss: 0.0691
+  Val Loss: 0.0980
+  Val PCK: 0.9914
+  Val IoU: 0.9430
+  No improvement for 20/20 epochs
+
+Early stopping triggered after 158 epochs!
+wandb: 
+wandb: Run history:
+wandb:            batch ▂▆█▅▅▅▆▁▄▅▇▅▃▄▆▂█▆▅▃█▅▅█▇█▄█▆▅▆▂▅▃▄▅▄▃▅▅
+wandb:            epoch ▁▁▁▁▁▂▂▂▂▂▂▂▃▃▃▃▃▃▃▄▄▄▄▄▄▅▅▅▅▅▆▆▆▆▆▇▇▇██
+wandb:               lr ▂███████▇▇▇▇▇▇▇▆▆▆▆▆▅▅▅▅▄▄▄▄▄▄▃▃▃▃▂▂▂▂▂▁
+wandb: model/parameters ▁
+wandb: train/epoch_loss █▄▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁
+wandb:    train/l_angle █▅▂▂▂▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁
+wandb:     train/l_diag ██▄▅▇▄▄▃▆▄▂▄▃▄▅▃▃▄▆▃▄▃▃▅▄▃▃▄▄▄▃▄▃▁▄▃▄▄▃▅
+wandb:     train/l_edge ▇▄▄▆▄▄▃▄▃▂▃▅█▂▂▂▃▃▂▂▃▂▂▂▃▂▁▂▂▂▂▁▂▁▂▁▂▁▂▂
+wandb:       train/l_hm █▂▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁
+wandb:      train/l_kpt █▂▂▁▁▁▁▁▂▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▁▂▁▁▁▁▁▁▁▁
+wandb:               +5 ...
+wandb:
+wandb: Run summary:
+wandb:            batch 2300
+wandb:       best_epoch 138
+wandb:    best_val_loss 0.09766
+wandb:            epoch 158
+wandb:               lr 2e-05
+wandb: model/parameters 1238856
+wandb: train/epoch_loss 0.06909
+wandb:    train/l_angle 0.01886
+wandb:     train/l_diag 0.00304
+wandb:     train/l_edge 0.00351
+wandb:               +7 ...
+wandb:
+wandb:  View run CourtKeyNet_01/21_22:34 at: https://wandb.ai/adithyanraj03/courtkeynet/runs/wijqjzd9
+wandb:  View project at: https://wandb.ai/adithyanraj03/courtkeynet
+wandb: Synced 5 W&B file(s), 159 media file(s), 2 artifact file(s) and 0 other file(s)
+wandb: Find logs at: .\wandb\run-20260121_223455-wijqjzd9\logs
+
+Training complete! Best val loss: 0.0977
+Weights saved to: runs\courtkeynet\exp_20260121_223500_wijqjzd9