replace_head {mlr3torch} | R Documentation |
Replace the head of a network Replaces the head of the network with a linear layer with d_out classes.
Description
Replace the head of a network Replaces the head of the network with a linear layer with d_out classes.
Usage
replace_head(network, d_out)
Arguments
network |
( |
d_out |
( |
[Package mlr3torch version 0.2.1 Index]