Pytorch, Optimization, Deep Learning, Artificial Intelligence, Tensorboard.
code block below we propose an alternative implementation of the PathDropout module forward function that produces the same output using torch.Tensor indexing instead of torch.gather. The modified lines of code have been highlighted. from timm.layers import PatchDropout class