search

BAAI-DCAI / SegVol

The official code for "SegVol: Universal and Interactive Volumetric Medical Image Segmentation".
MIT License
248 stars 22 forks source link

About pre-trained ViT weight file #26

Closed H47sj closed 2 weeks ago

H47sj commented 2 weeks ago

Hi, Thank you for your great job and sharing the models. Based on your paper's introduction:" ......to train the 3D ViT encoder of SegVol on the collected 96K CTs for 2000 epochs.......", I downloaded the weight file "ViT_pretrain.ckpt" in Google Drive. But I found its epoch num is 399 instead of 2000. So, may I ask if you can share the 2000 epochs' weight file? Thanks.

baifanxxx commented 2 weeks ago

Hi,

Now, the released weight is what we described in the paper. The 399 is not the overall epoch but the continual pre-trained epoch because we load the weights trained multiple times to continue the pre-training.

H47sj commented 2 weeks ago

Hi,

Now, the released weight is what we described in the paper. The 399 is not the overall epoch but the continual pre-trained epoch because we load the weights trained multiple times to continue the pre-training. Thanks for your reply. I understand.