.FastAI
ParamGroups
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
)
)
===
nothing
There are
3
methods for FastAI.ParamGroups
:
The following pages link back here:
training/discriminativelrs.jl , training/finetune.jl , training/paramgroups.jl