srmamamba-liver-segmentation / model_loader.py

Commit History

Fix inference pipeline: normalization, RAS orientation, state_dict check, debug logging
91a070d

Harshith Reddy commited on

Fix TF32 API
d8b8562

Harshith Reddy commited on

Fix: Exact import test format, TF32 set directly (no try/except), verify allocator config includes max_split_size_mb=128
5f56790

Harshith Reddy commited on

Fix: PyTorch 2.9.0 cu128, simplified setup.sh with fatal errors, TF32=tf32, cuDNN always on, TORCH_CUDA_ARCH_LIST=8.9
6bf725e

Harshith Reddy commited on

Fixed
ff6b142

Harshith Reddy commited on

Production build strategy: Docker support, pinned dependencies, OOM guardrails, optimized defaults
05125f7

Harshith Reddy commited on

Optimize torch.compile: default to reduce-overhead for faster first run, add compilation time warnings
b89934f

Harshith Reddy commited on

Speed optimizations: GPU aggregation, max-autotune compile, increased batch/ROI, GPU utilization monitoring, quality guardrails
59e6707

Harshith Reddy commited on

Fix OOM: reduce ROI depth to 64, sw_batch_size=1, CPU aggregation, fix env var timing, update TF32 API
8aecceb

Harshith Reddy commited on

Performance optimizations: warm-up, channels-last, TF32, GPU monitoring, memory-mapped I/O
c8db90d

Harshith Reddy commited on

Major performance and stability improvements: disable torch.compile by default, GPU preprocessing, ROI auto-adjustment, timeout mechanism, validation, cuDNN benchmarking, version logging, concurrency limits
700e0b1

Harshith Reddy commited on

Optimize for L40S GPU (48GB VRAM): increase window sizes and batch sizes for better performance
197d4e2

Harshith Reddy commited on

Fix checkpoint loading order: load before torch.compile to avoid _orig_mod prefix mismatch
ef22c5d

Harshith Reddy commited on

Add torch.compile optimization for better memory efficiency
f6ab828

Harshith Reddy commited on

Fix GPU OOM: Add post-load memory check, use ultra-minimal settings for low memory, and improve OOM recovery
11a9b37

Harshith Reddy commited on

Optimize for dedicated GPU: Add L4 GPU support with higher memory thresholds and better batch sizes
e1b9ddc

Harshith Reddy commited on

Fix naming conflict: rename model.py to model_loader.py and initialize DEVICE to prevent NoneType errors
493e2f1

Harshith Reddy commited on