Beckschen/3D-TransUNet

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

Opened this issue · 0 comments

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?