update

xinwucwp · May 6, 2020 · 56f0a45 · 56f0a45
1 parent cff7382
commit 56f0a45
Show file tree

Hide file tree

Showing 7 changed files with 3 additions and 1 deletion.
diff --git a/.train.py.swp b/.train.py.swp
diff --git a/.utils.py.swp b/.utils.py.swp
diff --git a/__pycache__/utils.cpython-37.pyc b/__pycache__/utils.cpython-37.pyc
diff --git a/log/training/events.out.tfevents.1588726772.ustc b/log/training/events.out.tfevents.1588726772.ustc
diff --git a/log/training/events.out.tfevents.1588727343.ustc b/log/training/events.out.tfevents.1588727343.ustc
diff --git a/log/validation/events.out.tfevents.1588727098.ustc b/log/validation/events.out.tfevents.1588727098.ustc
diff --git a/train.py b/train.py
@@ -26,6 +26,7 @@ def goTrain():
           'shuffle': True}
   seismPathT = "./data/train/seis/"
   faultPathT = "./data/train/fault/"
+
   seismPathV = "./data/validation/seis/"
   faultPathV = "./data/validation/fault/"
   train_ID = range(200)
@@ -35,7 +36,8 @@ def goTrain():
   valid_generator = DataGenerator(dpath=seismPathV,fpath=faultPathV,
                                   data_IDs=valid_ID,**params)
   model = unet(input_size=(None, None, None,1))
-  model.compile(optimizer=Adam(lr=1e-3), loss='binary_crossentropy')
+  model.compile(optimizer=Adam(lr=1e-3), loss='binary_crossentropy', 
+                metrics=['accuracy'])
   model.summary()
 
   # checkpoint