ML in an experimental science so I think pretty much can work, you just need to experiment.
A nn.Layer has weights and biases that are Tensors so you can basically assign stuff to a slice of the tensor weights[:,:num_classes] = old_weights