--- a
+++ b/config.json
@@ -0,0 +1,35 @@
+{
+  "mode" : "3d",
+  
+  "path" :
+    {
+      "scans" : "/powerai/data/retina-unet/data/LIDC-IDRI",
+      "masks" : "/powerai/data/retina-unet/data/lung_masks_LUNA16",
+      "labelled_list" : "/powerai/data/retina-unet/data/labelled.pickle"
+    },
+
+  "train3d" :
+    {
+      "epochs" : 10,
+      "batch_size" : 1,
+      "lr" : 1e-4,
+      "start_filters" : 32,
+      "scan_size" : [128, 256, 256],
+      "train_size" : 800,
+      "validation_steps" : 30,
+      "validation_size" : 50,
+      "n_classes" : 1
+    },
+
+  "train2d" :
+    {
+      "epochs" : 10,
+      "scans_per_batch" : 1,
+      "slices_per_batch" : 4,
+      "neg_examples_per_batch" : 0,
+      "lr" : 1e-4,
+      "start_filters" : 32,
+      "train_size" : 64
+    }
+}
+