replace_head: Replace the head of a network Replaces the head of the...
In mlr3torch: Deep Learning with 'mlr3'
replace_head 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.
mlr3torch documentation built on April 4, 2025, 3:03 a.m.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
| replace_head | 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 |
( |
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
Embedding an R snippet on your website
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.