2db50bdc87eac94438110fa0d895cbe269031eb2,train.py,,,#,5

Before Change

            [(32, 7), (32, 5)],
            [(64, 5), (64, 5)],
            [(128, 5), (128, 5)],
            [(128, 5), (128, 5)],
            [(128, 5), (128, 5)]
        ],
        "resized_size": (600, 400),

After Change

        model_params["num_classes"] = 1

    // Exporting params
    if not os.path.isdir(args["model_output_dir"]):
        os.mkdir(args["model_output_dir"])
    with open(os.path.join(args["model_output_dir"], "model_params.json"), "w") as f:
        json.dump(model_params, f)

    session_config = tf.ConfigProto()

Instances