additional logging statements

Author: rabowskyb

tf-2-workflow/train_model/train.py

@@ -49,6 +49,8 @@ def get_test_data(test_dir):
 
     args, _ = parse_args()
 
+    print('Training data location: {}'.format(args.train))
+    print('Test data location: {}'.format(args.test))
     x_train, y_train = get_train_data(args.train)
     x_test, y_test = get_test_data(args.test)
 
@@ -74,4 +76,3 @@ def get_test_data(test_dir):
         # save model
         model.save(args.model_dir + '/1')
 
-