UV - fast Python package installer and resolver

uv-monai-torchsummary.py

"""
>> pip install uv # OR https://docs.astral.sh/uv/getting-started/installation/
>> uv run -p 3.11 uv-monai-torchsummary.py
"""

# /// script
# requires-python = ">=3.11"
# dependencies = [
#     "monai==1.3.2",
#     "torchsummary==1.5.1",
# ]
# ///
import monai # v1.3.2
import torchsummary # v1.5.1
model = monai.networks.nets.UNet(in_channels=2, out_channels=1, spatial_dims=3, channels=[16, 32, 64, 128], strides=[2, 2, 2], num_res_units=2)
torchsummary.summary(model.cuda(), (2, 144, 144, 144))
print ('\n\n =========================== \n\n')
model = monai.networks.nets.UNet(in_channels=5, out_channels=1, spatial_dims=3, channels=[16, 32, 64, 128], strides=[2, 2, 2], num_res_units=2)
torchsummary.summary(model.cuda(), (5, 144, 144, 144))