DDMR / COMET /train_config_files /Config_SEGGUIDED_bottom_froozen.ini
jpdefrutos's picture
Scripts for training on the COMET CT Dataset
476daa5
raw
history blame contribute delete
697 Bytes
[LOSSES]
segmentation = dice
similarity = ssim
[TRAIN]
model = /mnt/EncryptedData1/Users/javier/train_output/Brain_study/No_gamma/SEGGUIDED_Lsim_ncc__Lseg_dice__MET_mse_ncc_ssim_013319-11092021/checkpoints/best_model.h5
batchSize = 8
learningRate = 1e-5
accumulativeGradients = 1
gpu = 2
regularizationWeight = 1e-5
epochs = 10000
name = SEGGUIDED
freeze = BOTTOM
[DATASETS]
train = /mnt/EncryptedData1/Users/javier/ext_datasets/COMET_dataset/OSLO_COMET_CT/Formatted_128x128x128/train
validation = /mnt/EncryptedData1/Users/javier/ext_datasets/COMET_dataset/OSLO_COMET_CT/Formatted_128x128x128/validation
[OTHERS]
outputFolder = /mnt/EncryptedData1/Users/javier/train_output/COMET/BOTTOM_FROZEN