ad2ff3124948b32978b41d5b3e66630f05d0f6fe,beholder_demo.py,,,#,210

Before Change

if __name__ == "__main__":
  parser = argparse.ArgumentParser()
  parser.add_argument("--fake_data", nargs="?", const=True, type=bool,
                      default=False,
                      help="If true, uses fake data for unit testing.")
  parser.add_argument("--max_steps", type=int, default=10000,
                      help="Number of steps to run trainer.")
  parser.add_argument("--learning_rate", type=float, default=0.001,
                      help="Initial learning rate")

After Change

if __name__ == "__main__":
    import os
    if not os.path.exists(LOG_DIRECTORY):
        os.makedirs(LOG_DIRECTORY)
    print(LOG_DIRECTORY)
    beholder_pytorch()

Instances