Commit c33ebc04 authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #305 from miaecle/kaggle 

Fix for benchmark running scripts
parents f95d82c8 d7d5b027

examples/run_all_benchmarks.py

+1 −1
Original line number Diff line number Diff line
@@ -36,6 +36,6 @@ hps['graphconv'] = [{'learning_rate': 0.001, 'n_filters': 64, 
for model in models:

  for dataset in datasets:

    print("Benchmarking %s on dataset %s" % (model, dataset))

    benchmark_loading_datasets(base_dir, hps, dataset_name=dataset,

    benchmark_loading_datasets(base_dir, hps, dataset=dataset,

                               model=model, reload=True,

                               verbosity='high', out_path=out_path)

examples/run_benchmark.py

+1 −1
Original line number Diff line number Diff line
@@ -108,7 +108,7 @@ for i in range(int(sys.argv[3])): 
        f.write(item+',')

        f.write(str(hps[model][i][item])+',')



benchmark_loading_datasets(base_dir_o, hps, dataset_name=dname,

benchmark_loading_datasets(base_dir_o, hps, dataset=dname,

                           model=model, reload=True,

                           verbosity='high', out_path=out_path)

CRA Git | Maintained and supported by SUSTech CRA and CCSE