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 |
(torch::nn_module )
The network
|
d_out |
(integer(1) )
The number of output classes.
|
[Package
mlr3torch version 0.1.1
Index]