Update parallel_model.py

parallel_model.py

@@ -1,18 +1,3 @@
-"""
-Mask R-CNN
-Multi-GPU Support for Keras.
-
-Copyright (c) 2017 Matterport, Inc.
-Licensed under the MIT License (see LICENSE for details)
-Written by Waleed Abdulla
-
-Ideas and a small code snippets from these sources:
-https://github.com/fchollet/keras/issues/2436
-https://medium.com/@kuza55/transparent-multi-gpu-training-on-tensorflow-with-keras-8b0016fd9012
-https://github.com/avolkov1/keras_experiments/blob/master/keras_exp/multigpu/
-https://github.com/fchollet/keras/blob/master/keras/utils/training_utils.py
-"""
-
 import tensorflow as tf
 import keras.backend as K
 import keras.layers as KL
@@ -20,12 +5,6 @@ import keras.models as KM
 
 
 class ParallelModel(KM.Model):
-    """Subclasses the standard Keras Model and adds multi-GPU support.
-    It works by creating a copy of the model on each GPU. Then it slices
-    the inputs and sends a slice to each copy of the model, and then
-    merges the outputs together and applies the loss on the combined
-    outputs.
-    """
 
     def __init__(self, keras_model, gpu_count):
         """Class constructor.
@@ -172,4 +151,4 @@ if __name__ == "__main__":
         validation_data=(x_test, y_test),
         callbacks=[keras.callbacks.TensorBoard(log_dir=MODEL_DIR,
                                                write_graph=True)]
-    )
+    )