.FastAIParamGroups
struct defined in module
FastAI
ParamGroups(grouper, m)
A logical grouping of parameters in
m created by
ParamGrouper
grouper. Parameters in
m are assigned a group that can be queried using
getgroup(paramgroups, param). If a parameter is not assigned a group,
getgroup returns
nothing.
using
Flux
:
Chain
,
Dense
,
params
using
FastAI
:
ParamGroups
,
IndexGrouper
,
getgroup
model
=
Chain
(
Dense
(
3
,
5
)
,
Dense
(
5
,
3
)
)
paramgroups
=
ParamGroups
(
IndexGrouper
(
[
1
,
2
]
)
,
model
)
getgroup
(
paramgroups
,
model
[
1
]
.
weight
)
==
1
getgroup
(
paramgroups
,
model
[
2
]
.
weight
)
==
2
getgroup
(
paramgroups
,
rand
(
10
)
)
===
nothingThere are
3
methods for FastAI.ParamGroups:
The following pages link back here:
training/discriminativelrs.jl , training/finetune.jl , training/paramgroups.jl