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.2 Index]