Replace the head of a network Replaces the head of the network with a linear layer with d_out classes.
Useful links