Sistem Nn Not Trainee

For example in iniit we configure different trainable layers including convolution and affine layers with nn conv2d and nn linear respectively