Beckschen / 3D-TransUNet

This is the official repository for the paper "3D TransUNet: Advancing Medical Image Segmentation through Vision Transformers"
Apache License 2.0
201 stars 13 forks source link

How to train the three configurations - encoder only, decoder only, encoder plus decoder? #30

Open stelatopalova opened 6 months ago

stelatopalova commented 6 months ago

Hello,

I want to train your architecture using my own dataset. I already preprocessed the dataset using nnunet and made a config.yaml file using the examples you provided for the three datasets here. I set the hdfs_base parameter to Task511_3DTransUNet_encoder_only with Task511 being my own dataset. However, from the code in train.py I see that this hdfs_base parameter is only used in get_default_configuration() for the name of the folder. Is that the only parameter I need to set in order to train the three different configurations - encoder only, decoder only and encoder plus decoder?