Initialize weights to sane defaults

def reset_weights(self):
        """ Initialize weights to sane defaults """
        init.uniform_(self.linear.weight, -3e-3, 3e-3)
        init.zeros_(self.linear.bias)