## Training procedure ### Framework versions - SWIFT 1.5.4 ### Base model information - BaseModel Class UNet3DConditionModel