Skip to content
Permalink
Browse files

removed clean flag

  • Loading branch information
rusty1s committed Feb 19, 2020
1 parent 09f9ada commit f3e30a3dde686a36c2db5b1c41742a53925bf52b
Showing with 1 addition and 2 deletions.
  1. +1 −2 benchmark/kernel/main.py
@@ -59,8 +59,7 @@ def logger(info):
best_result = (float('inf'), 0, 0) # (loss, acc, std)
print('-----\n{} - {}'.format(dataset_name, Net.__name__))
for num_layers, hidden in product(layers, hiddens):
dataset = get_dataset(dataset_name, sparse=Net != DiffPool,
cleaned=True)
dataset = get_dataset(dataset_name, sparse=Net != DiffPool)
model = Net(dataset, num_layers, hidden)
loss, acc, std = cross_validation_with_val_set(
dataset,

0 comments on commit f3e30a3

Please sign in to comment.
You can’t perform that action at this time.