DDMR / COMET /train_config_files /Config_UW_bottom_froozen.ini
jpdefrutos's picture
Scripts for training on the COMET CT Dataset
476daa5
raw
history blame contribute delete
696 Bytes
[LOSSES]
segmentation = dice,hd
similarity = ncc,ssim
[TRAIN]
model = /mnt/EncryptedData1/Users/javier/train_output/Brain_study/No_gamma/UW_Lsim_ncc__ssim__Lseg_dice__MET_mse_ncc_ssim_204557-02092021/checkpoints/best_model.h5
batchSize = 8
learningRate = 1e-5
accumulativeGradients = 1
gpu = 2
regularizationWeight = 1e-5
epochs = 10000
name = UW
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