Use torch.amax
instead of torch.max
#5224
Open
torch.amax
instead of torch.max
#5224