Model Summary              
┏━━━━━━━━━━━━━━━━━━━━━━┳━━━━━━━━━━━━━━━┓
┃      Parameter       ┃     Value     ┃
┡━━━━━━━━━━━━━━━━━━━━━━╇━━━━━━━━━━━━━━━┩
│       Network        │ monai_attunet │
│ Trainable Parameters │    1987434    │
│   Output channels    │       2       │
└──────────────────────┴───────────────┘