cosmicman-cvpr2024 / CosmicMan

CosmicMan: A Text-to-Image Foundation Model for Humans (CVPR 2024)
317 stars 8 forks source link

An error about AspectBucket #7

Closed ChaoFan996 closed 4 months ago

ChaoFan996 commented 4 months ago

Hi, thanks for your excellent work! I have tried to train the model but met an error: `

rank4: File "./CosmicMan/dataset.py", line 334, in init_buckets rank4: self._bucket_interp = interp1d(self._bucket_ratios, list(range(len(buckets))), assume_sorted=True, rank4: File "/root/miniconda/envs/cosmicman/lib/python3.10/site-packages/scipy/interpolate/_interpolate.py", line 543, in init rank4: self._y = self._reshape_yi(self.y) rank4: File "/root/miniconda/envs/cosmicman/lib/python3.10/site-packages/scipy/interpolate/_polyint.py", line 113, in _reshape_yi rank4: return yi.reshape((yi.shape[0], -1)) ` It seems that the self.buckets has been wrongly set to an empty list. Would you provide any suggestions to address this issue? Thanks!

ChaoFan996 commented 4 months ago

I may fix this bug. The arg name in this line should be bucket_side_max rather than bucket_side_min, right?

leesky1c commented 4 months ago

Yep! Thanks for the correction XD