forked from mlpack/benchmarks
-
Notifications
You must be signed in to change notification settings - Fork 0
/
small_config.yaml
35 lines (34 loc) · 870 Bytes
/
small_config.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
# Block for general settings.
library: general
settings:
# Time until a timeout in seconds.
timeout: 9000
database: 'reports/benchmark.db'
keepReports: 20
---
# MLPACK:
# A Scalable C++ Machine Learning Library
library: mlpack
methods:
MLP_FORWARD:
run: ['metric']
iteration: 3
script: methods/mlpack/mlp_forward.py
format: [csv, txt]
datasets:
- files: ['None']
options:
input_size: 50000
hidden_size: 5000
output_size: 100
MLP_BACKWARD:
run: ['metric']
iteration: 3
script: methods/mlpack/mlp_backward.py
format: [csv, txt]
datasets:
- files: ['None']
options:
input_size: 50000
hidden_size: 5000
output_size: 100