Update train.py

Update imagenet pretrain and delete heads training

train.py

@@ -28,7 +28,7 @@
                           model_dir=MODEL_DIR)
 
 # Which weights to start with?
-init_with = "coco"  # imagenet, coco, or last
+init_with = "imagenet"  # imagenet, coco, or last
 
 if init_with == "imagenet":
     model.load_weights(model.get_imagenet_weights(), by_name=True)
@@ -57,12 +57,12 @@
 # Passing layers="heads" freezes all layers except the head
 # layers. You can also pass a regular expression to select
 # which layers to train by name pattern.
-model.train(dataset_train, dataset_val, 
-            learning_rate=bowl_config.LEARNING_RATE, 
-            epochs=1, 
-            layers='heads')
+#model.train(dataset_train, dataset_val, 
+#            learning_rate=bowl_config.LEARNING_RATE, 
+#            epochs=1, 
+#            layers='heads')
 
 model.train(dataset_train, dataset_val, 
             learning_rate=bowl_config.LEARNING_RATE / 10,
             epochs=100, 
-            layers="all")
+            layers="all")