Table 2 Experimental environment settings.
From: VMKLA-UNet: vision Mamba with KAN linear attention U-Net
Operating system | Linux Ubuntu 22.04 |
|---|---|
Python Version | 3.8 |
Framework and CUDA version | Torch 2.1.0 CUDA 12.1 |
Graphics | NVIDIA V100 Tensor Core 32G * 1 |
Epoch | 300 |
Batch size | 32 |
Learning rate | 1e − 4 |
Learning strategy | Cosine Annealing |
Optimizer | AdamW |
