YVIq2
Hyperparameter interface
abstract
type
HyperParameter
{
T
}
end
function
sethyperparameter!
end
stateaccess
(
::
Type
{
HyperParameter
}
)
=
(
)
Implementations
abstract
type
LearningRate
<:
HyperParameter
{
Float64
}
end
stateaccess
(
::
Type
{
LearningRate
}
)
=
(
optimizer
=
Write
(
)
,
)
function
sethyperparameter!
(
learner
,
::
Type
{
LearningRate
}
,
value
)
learner
.
optimizer
=
setlearningrate!
(
learner
.
optimizer
,
value
)
return
learner
end
function
setlearningrate!
(
optimizer
::
Flux
.
Optimise
.
AbstractOptimiser
,
value
)
optimizer
.
eta
=
value
optimizer
end
function
setlearningrate!
(
optimizer
,
value
)
@
set
optimizer
.
eta
=
value
end