17 lines
295 B
YAML
17 lines
295 B
YAML
program: train.py
|
|
method: grid
|
|
metric:
|
|
name: valid.total
|
|
goal: minimize
|
|
parameters:
|
|
train_params.batch_size:
|
|
values: [8]
|
|
train_params.focal_gamma:
|
|
values: [0, 1]
|
|
nn_params.main_decoder.input_length:
|
|
values: [8192]
|
|
|
|
command:
|
|
- python3
|
|
- ${program}
|
|
- ${args_no_hyphens} |