optimise.py 文件源码

def TRAIN(args, model):
    t0 = time.time()
    print("\nParameters:")
    for attr, value in sorted(vars(args).items()):
        print("{}={}".format(attr.upper(), value))
    print()
    print("Graph initialized..")
    t1 = time.time()
    print("time taken:", t1-t0)
    print()
    data = load_data(args, args.data, saved=args.load_data)
    run_network(args, data, model, tuning=args.tune)