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