fix(train): fix save model when using fsdp

ibrahimethemhamamci / CT-CLIP

Developing Generalist Foundation Models from a Multimodal Dataset for 3D Computed Tomography

200 stars 21 forks source link

Open wangxiaoyuwangdayu opened 6 months ago

wangxiaoyuwangdayu commented 6 months ago

Fixed the issue that the saved state_dict was incomplete when using fsdp