Fix UNet architecture: add norm='batch' parameter and use strict=False for state_dict loading 1338ad7 verified IFMedTechdemo commited on Oct 1