Hi, Can anyone please guide how can we add the regularizer in the nn.Linear module same like tensorflow. In Tensorflow we add regularization in the Dense layer as:
tf.layers.dense(inputs=pool2_flat, units=2048, name='dense1', kernel_regularizer=tf.contrib.layers.l2_regularizer(0.001), bias_regularizer=tf.contrib.layers.l2_regularizer(0.001))
Any thought how can add in nn.Linear?
Thank you.