Closed Mike001-wq closed 2 months ago
Should lines 234 and 235 in utils/general_utils.py be swapped? First record the val with the lowest probability, then replace it with the value with the highest probability. `
weight[weight == val] = unique_vals[counts.argmax()] self.tail_locs[val.item()] = torch.where(weight == val)[0].detach().cpu()`
Thanks for highlighting this bug. It is now fixed in a recent commit (1cfc84d). Feel free to reopen if there are any issues.
Should lines 234 and 235 in utils/general_utils.py be swapped? First record the val with the lowest probability, then replace it with the value with the highest probability. `