给的train_cls train_bbs的维度是256，[GPSDiffusion-Object-Shadow-Generation-SDXL](https://github.com/bcmi/GPSDiffusion-Object-Shadow-Generation-SDXL/tree/9a4b0a93439d74a9a61654ac3a9149d0823d1f54)里训练和Infer都是512
