0fb615b9a68f6eb642598b360c007a64d022027a,examples/pytorch/ogb/deepwalk/deepwalk.py,,,#,236

Before Change

    parser.add_argument("--only_gpu", default=False, action="store_true", 
            help="training with GPU")

    parser.add_argument("--adam", default=False, action="store_true", 
            help="use adam for embedding updation, recommended")
    parser.add_argument("--sgd", default=False, action="store_true", 
            help="use sgd for embedding updation")
    parser.add_argument("--avg_sgd", default=False, action="store_true", 
            help="average gradients of sgd for embedding updation")

After Change

    args = parser.parse_args()

    if args.async_update:
        assert args.mix, "--async_update only with --mix"

    start_time = time.time()
    trainer = DeepwalkTrainer(args)
    trainer.train()
    print("Total used time: %.2f" % (time.time() - start_time))

Instances