prepare_feat_model[source]

prepare_feat_model(m)

get_layers[source]

get_layers(m, idxs)

class FeatModels[source]

FeatModels()

class LayerFeats[source]

LayerFeats(m, stl_ls, cnt_ls, tfms=None) :: Module

Same as nn.Module, but no need for subclasses to call super().__init__

test_eq(len(stl_fts), 5)
test_eq(len(cnt_fts), 1)