Implemented weight saving callback in model_training.py

visualizations/train_viz_data_generator.py

@@ -100,8 +100,6 @@ parser.add_argument(
     "-hp",
     help="Path pointing to a pickled dictionary of network hyperparameters. "
     "Thought to be used with the output of hyperparameter_tuning.py",
-    default=None,
-    type=str,
 )
 parser.add_argument(
     "--encoding-size",