Skip to content
Permalink
Browse files

linting

  • Loading branch information...
rusty1s committed Oct 28, 2019
1 parent d393b2d commit beda36b281b350793f918c9bb1a6eb6476423316
Showing with 10 additions and 4 deletions.
  1. +10 −4 benchmark/kernel/main.py
@@ -61,11 +61,17 @@ def logger(info):
cleaned=True)
model = Net(dataset, num_layers, hidden)
loss, acc, std = cross_validation_with_val_set(
dataset, model, folds=10, epochs=args.epochs,
batch_size=args.batch_size, lr=args.lr,
dataset,
model,
folds=10,
epochs=args.epochs,
batch_size=args.batch_size,
lr=args.lr,
lr_decay_factor=args.lr_decay_factor,
lr_decay_step_size=args.lr_decay_step_size, weight_decay=0,
logger=None)
lr_decay_step_size=args.lr_decay_step_size,
weight_decay=0,
logger=None,
)
if loss < best_result[0]:
best_result = (loss, acc, std)

0 comments on commit beda36b

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