fdeeaff5e452e257bf3a3dd76e25c81304cc4c98,mmdnn/conversion/examples/tensorflow/vis_meta.py,,_main,#,14

Before Change

    

    if len(sys.argv) != 3:
        raise ValueError("Usage: python vis_meta.py /path/to/model.meta /path/to/log/directory")
    // load file
    visualize(sys.argv[1], sys.argv[2])
    os.system("tensorboard --logdir=" + sys.argv[2])

After Change

    args, unknown_args = parser.parse_known_args()

    if not os.path.isfile(args.ckpt):
        print("The checkpoint meta file does not exist.")
        exit(1)

    if not os.path.isdir(args.logdir):
        print("The log directory does not exist.")

Instances