预训练CXR-BERT 和 ConvNeXt路径设置

Junelin2333 / LanGuideMedSeg-MICCAI2023

Pytorch code of MICCAI 2023 Paper-Ariadne’s Thread : Using Text Prompts to Improve Segmentation of Infected Areas from Chest X-ray images

GNU General Public License v3.0

32 stars 1 forks source link

预训练CXR-BERT 和 ConvNeXt路径设置 #6

Closed WDYTX closed 6 months ago

WDYTX commented 6 months ago

这几个在哪里啊？

Junelin2333 commented 6 months ago

Just change the bert_type and vision_type in /config/training.yaml to local filefold path.

Don't forget Star this repo and cite the paper.（白嫖哒咩） Best wishes :)

WDYTX commented 6 months ago

Thank you very much for your reply. I have one more question. Can I run without an NVIDIA Tesla V100 SXM3 without an NVIDIA Tesla V100 SXM3 32GB VRAM GPU? Is 3090 or 2080ti OK?

Junelin2333 commented 6 months ago

Nvidia RTX 3090 24GB VRAM is ok. Also 2080Ti. The VRAM occupation of the method is about less than 10GB with batchsize 32.

Hope this reply could help you :)