Update seg_lesion_train.py

imatge-upc · Nov 29, 2017 · e8ccc99 · e8ccc99
1 parent 8837c53
commit e8ccc99
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/seg_lesion_train.py b/seg_lesion_train.py
@@ -32,10 +32,10 @@
 values = [ini_learning_rate, ini_learning_rate * 0.1, ini_learning_rate, ini_learning_rate * 0.1, ini_learning_rate,
           ini_learning_rate * 0.1]
 
-task = 'seg_lesion'
+task_name = 'seg_lesion'
 
 database_root =  os.path.join(root_folder, 'LiTS_database')
-logs_path = os.path.join(root_folder, 'train_files', task, 'networks')
+logs_path = os.path.join(root_folder, 'train_files', task_name, 'networks')
 imagenet_ckpt = os.path.join(root_folder, 'train_files', 'vgg_16.ckpt')
 
 train_file = os.path.join(root_folder, 'seg_DatasetList', 'training_lesion_commonbb_nobackprop_3.txt')