I used .detach() in gdas/gdas.py at main · promach/gdas · GitHub 1
self.chosen_edge = np.argmax(gumbel.detach().numpy(), axis=0)  # converts one-hot encoding into integer
How do I get around without using detach() ?