Skip to content

How to setting learning rate in the model params?? #842

@gaoliming123

Description

@gaoliming123

such as model build, where we can set the learning rate

model = mz.models.MatchPyramid()
model.params.update(preprocessor.context)
model.params['task'] = ranking_task
model.params['embedding_output_dim'] = 128
model.params['embedding_input_dim'] = preprocessor.context['embedding_input_dim']
model.params['embedding_trainable'] = True
model.params['num_blocks'] = 2
model.params['kernel_count'] = [8, 16]
model.params['kernel_size'] = [[5, 5], [3, 3]]
model.params['dpool_size'] = [3, 3]
model.params['optimizer'] = 'adam'
model.params['dropout_rate'] = 0.3
model.build()
model.compile()

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions