Improving Discrete Optimisation Via Decoupled Straight-Through Gumbel-Softmax